← all repositories

IvanDrokin/torch-conv-kan

A PyTorch collection of convolutional Kolmogorov-Arnold network layers and architectures with ImageNet/CIFAR benchmarks.

531 stars Python ML FrameworksComputer Vision
torch-conv-kan
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

This repository provides PyTorch implementations of 1D, 2D, and 3D convolutional layers using Kolmogorov-Arnold network theory, including variants such as KAN, Fast KAN, ChebyKAN, WavKAN, JacobiKAN, and BernsteinKAN. It offers complete model architectures like ResNet-like, DenseNet-like, VGG-like, and U-Net variants, along with training scripts using the accelerate library and pretrained weights for ImageNet1k. Benchmarks are provided on MNIST, CIFAR-10, CIFAR-100, TinyImageNet, and ImageNet1k datasets.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.