/jax-kan

Kolmogorov Arnold Networks in JAX, with extensions

Primary LanguageJupyter NotebookMIT LicenseMIT

Watchers