This is a PyTorch/GPU implementation of the paper Kolmogorov–Arnold Transformer (KAT), which replace the MLP layers in vision transformer with KAN layers.
forked from Adamdad/kat
-
Notifications
You must be signed in to change notification settings - Fork 0
Kolmogorov-Arnold Transformer: A PyTorch Implementation with CUDA kernel
License
Junkai03/kat
About
Kolmogorov-Arnold Transformer: A PyTorch Implementation with CUDA kernel
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published
Languages
- Jupyter Notebook 76.7%
- Python 22.6%
- Shell 0.7%