├── .gitignore ├── LICENSE.txt ├── README.md ├── benchmark.py ├── cu_fasterkan ├── __init__.py └── cuda.py ├── cuda ├── __init__.py ├── cpp │ ├── faster.cpp │ └── faster_cuda.cu └── setup.py ├── draw_spline_basis.ipynb ├── efficient_kan ├── __init__.py └── kan.py ├── fasterkan ├── __init__.py ├── fasterkan.py ├── fasterkan_basis.py ├── fasterkan_layers.py └── feature_extractor.py ├── fastkan ├── __init__.py └── fastkan.py ├── img ├── compare_basis.png └── val_curves.png ├── pdm.lock ├── pyproject.toml ├── test_running_time.ipynb ├── tests └── test_simple_math.py ├── times.txt ├── torchkan ├── KALnet.py ├── KANvolver.py ├── __init__.py ├── mnist_KANvolve.py └── wandb │ ├── run-20240513_230021-s01vwjqq │ ├── files │ │ └── config.yaml │ └── run-s01vwjqq.wandb │ └── run-20240513_230251-9qqgt0u7 │ ├── files │ └── config.yaml │ └── run-9qqgt0u7.wandb ├── train_cifar10.py └── train_mnist.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/README.md -------------------------------------------------------------------------------- /benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/benchmark.py -------------------------------------------------------------------------------- /cu_fasterkan/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cu_fasterkan/cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/cu_fasterkan/cuda.py -------------------------------------------------------------------------------- /cuda/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cuda/cpp/faster.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/cuda/cpp/faster.cpp -------------------------------------------------------------------------------- /cuda/cpp/faster_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/cuda/cpp/faster_cuda.cu -------------------------------------------------------------------------------- /cuda/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/cuda/setup.py -------------------------------------------------------------------------------- /draw_spline_basis.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/draw_spline_basis.ipynb -------------------------------------------------------------------------------- /efficient_kan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/efficient_kan/__init__.py -------------------------------------------------------------------------------- /efficient_kan/kan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/efficient_kan/kan.py -------------------------------------------------------------------------------- /fasterkan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/fasterkan/__init__.py -------------------------------------------------------------------------------- /fasterkan/fasterkan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/fasterkan/fasterkan.py -------------------------------------------------------------------------------- /fasterkan/fasterkan_basis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/fasterkan/fasterkan_basis.py -------------------------------------------------------------------------------- /fasterkan/fasterkan_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/fasterkan/fasterkan_layers.py -------------------------------------------------------------------------------- /fasterkan/feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/fasterkan/feature_extractor.py -------------------------------------------------------------------------------- /fastkan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/fastkan/__init__.py -------------------------------------------------------------------------------- /fastkan/fastkan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/fastkan/fastkan.py -------------------------------------------------------------------------------- /img/compare_basis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/img/compare_basis.png -------------------------------------------------------------------------------- /img/val_curves.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/img/val_curves.png -------------------------------------------------------------------------------- /pdm.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/pdm.lock -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/pyproject.toml -------------------------------------------------------------------------------- /test_running_time.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/test_running_time.ipynb -------------------------------------------------------------------------------- /tests/test_simple_math.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/tests/test_simple_math.py -------------------------------------------------------------------------------- /times.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/times.txt -------------------------------------------------------------------------------- /torchkan/KALnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/torchkan/KALnet.py -------------------------------------------------------------------------------- /torchkan/KANvolver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/torchkan/KANvolver.py -------------------------------------------------------------------------------- /torchkan/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/torchkan/__init__.py -------------------------------------------------------------------------------- /torchkan/mnist_KANvolve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/torchkan/mnist_KANvolve.py -------------------------------------------------------------------------------- /torchkan/wandb/run-20240513_230021-s01vwjqq/files/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/torchkan/wandb/run-20240513_230021-s01vwjqq/files/config.yaml -------------------------------------------------------------------------------- /torchkan/wandb/run-20240513_230021-s01vwjqq/run-s01vwjqq.wandb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/torchkan/wandb/run-20240513_230021-s01vwjqq/run-s01vwjqq.wandb -------------------------------------------------------------------------------- /torchkan/wandb/run-20240513_230251-9qqgt0u7/files/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/torchkan/wandb/run-20240513_230251-9qqgt0u7/files/config.yaml -------------------------------------------------------------------------------- /torchkan/wandb/run-20240513_230251-9qqgt0u7/run-9qqgt0u7.wandb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/torchkan/wandb/run-20240513_230251-9qqgt0u7/run-9qqgt0u7.wandb -------------------------------------------------------------------------------- /train_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/train_cifar10.py -------------------------------------------------------------------------------- /train_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AthanasiosDelis/faster-kan/HEAD/train_mnist.py --------------------------------------------------------------------------------