srigas / jaxKANLinks
Kolmogorov-Arnold Networks built on JAX
☆88Updated 6 months ago
Alternatives and similar repositories for jaxKAN
Users that are interested in jaxKAN are comparing it to the libraries listed below
Sorting:
- Efficient Differentiable n-d PDE solvers in JAX.☆47Updated last week
- TorchFSM: Fourier Spectral Method with PyTorch☆50Updated this week
- Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy☆92Updated this week
- An unofficial implementation of the Fourier Neural Operator in Flax☆18Updated last year
- Neural Emulator Architectures in JAX.