srkan 0.1.0

Spiking Rational Kolmogorov-Arnold Network

srKAN

Spiking Rational Kolmogorov-Arnold Network.

Installation

pip install srkan

Usage

import torch
from srkan import SRKAN

# Create a network
net = SRKAN(
    sizes=[8, 16, 5],
    order=4, 
    dt=1e-3, 
    rank=4, 
    gated=True,
    use_norm=True
)

# Dummy input
x = (torch.rand(8, 64, 8) < 0.3).float()
out = net(x)