Kolmogorov Arnold Networks Python Packages

avitai-artifex

Generative modeling for JAX/Flax NNX: VAEs, GANs, DDPM/DDIM/score/DiT/latent diffusion, normalizing flows, energy-based, autoregressive and geometric (SE(3), protein, point cloud, mesh) models; image, text, audio, molecular, protein, tabular and time-series modalities; a shared Trainer with callbacks and extensions; evaluation through calibrax.

torchcurves

Parametric differentiable curves with PyTorch for continuous embeddings, shape-restricted models, or KANs

jaxkan

Kolmogorov–Arnold Networks built on JAX

kan-gpt

The PyTorch implementation of Generative Pre-trained Transformers (GPTs) using Kolmogorov-Arnold Networks (KANs) for language modeling

kanx

One library, four surfaces. Production-grade Kolmogorov-Arnold Networks || TensorFlow + PyTorch + ONNX. || A small KAN beats a 10× larger MLP on smooth, separable target. One library. Two backends. Real ONNX export. Docker + Kubernetes ready.

qkan

PyTorch implementation of QKAN "Quantum-inspired Kolmogorov-Arnold Network" https://arxiv.org/abs/2509.14026

coxkan

CoxKAN: Kolmogorov-Arnold Networks for Interpretable, High-Performance Survival Analysis

kanditioned

LinearKAN: A very fast implementation of Kolmogorov-Arnold Networks

torchkan

PyTorch Kolmogorov-Arnold Networks (spline, Legendre, Chebyshev, convolutional) with quantisation, CUDA, and Integrated-Gradients interpretability.

timekan

Python library designed to integrate Kolmogorov Arnold Networks with recurrent mechanisms.

torchortho

[ICLR 2026] Polynomial, trigonometric, and tropical activations