This repository contains the implementation of the KAN-GSS-LM in pytorch.
The KAN-GSS-LM combines these two components Kolmogorov-Arnold Network and Gated State Spaces.
Take a look at the train.py
file, which includes a training and generation example.
edwko / kan-gated-state-spaces-lm Goto Github PK
View Code? Open in Web Editor NEWKolmogorov Arnold Network Gated State Spaces Language Model
License: Apache License 2.0