Kolmogorov–Arnold Networks built on JAX
☆97Feb 3, 2026Updated last month
Alternatives and similar repositories for jaxKAN
Users that are interested in jaxKAN are comparing it to the libraries listed below
Sorting:
- A fast, differentiable, and extensible public BBN code☆15Feb 13, 2026Updated 3 weeks ago
- ☆17Aug 13, 2024Updated last year
- Code for experiments on transformers using Markovian data.☆22Nov 22, 2024Updated last year
- A python package for Weak Lensing Implicit Inference with Differentiable Simulator☆17Mar 5, 2024Updated 2 years ago
- Tool to generate fake galaxy catalogs with realistic positions, morphologies and fluxes from the far-ultraviolet to the far-infrared.☆21Feb 2, 2024Updated 2 years ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Mar 10, 2024Updated last year
- coded with and corrected by Google Anti-Gravity☆13Nov 23, 2025Updated 3 months ago
- Simulation-based inference in JAX☆45Nov 24, 2025Updated 3 months ago
- Discovering Differential Equations with Physics-Informed Neural Networks and Symbolic Regression☆11Jul 28, 2023Updated 2 years ago
- JAX-accelerated nuclear equation of state inference and TOV solvers☆10Updated this week
- Physics Informed Sparse Identification of Nonlinear Dynamics☆12Jan 3, 2025Updated last year
- Generalized sparse regression for continuous and discrete data☆12Feb 23, 2026Updated last week
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆39Jan 16, 2026Updated last month
- A small library of paramaterizations and parameter constraints for PyTrees.☆50Jan 23, 2026Updated last month
- JAX reimplementation and bindings for the NVIDIA cuDecomp library☆54Feb 12, 2026Updated 3 weeks ago
- All-KAN: All Kolmogorov-Arnold Networks We Know☆30Feb 28, 2026Updated last week
- Differentiable Likelihood for CMB Analysis☆27Jan 21, 2026Updated last month
- Tackling the Curse of Dimensionality with Physics-Informed Neural Networks☆15May 17, 2024Updated last year
- JAX code for modeling nearly-Keplerian orbits☆17Jan 8, 2026Updated last month
- An automatic differentiation system for dense and sparse problems☆13Jan 16, 2025Updated last year
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆13May 23, 2024Updated last year
- SE-PINN: Solving the Schrödinger Equation via Physics-Informed Machine Learning☆11Dec 17, 2025Updated 2 months ago
- ☆406Nov 14, 2025Updated 3 months ago
- Unitful Quantities in JAX☆53Updated this week
- Julia implementation of the Kolmogorov-Arnold network with custom gradients for fast training.☆85Feb 7, 2026Updated last month
- A code for velocity-based Lagrangian and Eulerian PT expansions of redshift-space distortions.☆25Jul 11, 2025Updated 7 months ago
- Introductory workshop on using deep neural networks for symbolic regression☆16Feb 8, 2026Updated 3 weeks ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆255Oct 28, 2025Updated 4 months ago
- [NeurIPS 2023 Spotlight] Separable Physics-Informed Neural Networks☆89Oct 25, 2023Updated 2 years ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆521Updated this week
- Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs + Adaptative Loss Weights with JAX 📓 Check out our various notebooks to g…☆45Updated this week
- ☆219Feb 21, 2026Updated 2 weeks ago
- JAX-powered Cosmological Particle-Mesh N-body Solver☆59Feb 24, 2026Updated last week
- ☆273Updated this week
- Precise symbolic emulators of the linear and nonlinear matter power spectrum☆15Oct 22, 2025Updated 4 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆13Jun 12, 2024Updated last year
- Stochastic Gradient MCMC for Jax☆18May 19, 2025Updated 9 months ago
- Repository for xL-SINDy, a robust algorithm to extract Lagrangian of nonlinear dynamical systems from noisy measurement data.☆11Aug 8, 2022Updated 3 years ago
- JAX implementation of Kolmogorov Arnold Networks (KANs).☆10May 7, 2024Updated last year