srigas / jaxKANView external linksLinks
Kolmogorov–Arnold Networks built on JAX
☆96Feb 3, 2026Updated last week
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☆15Jan 28, 2026Updated 2 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 last year
- 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
- coded with and corrected by Google Anti-Gravity☆13Nov 23, 2025Updated 2 months ago
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆20Mar 10, 2024Updated last year
- Simulation-based inference in JAX☆45Nov 24, 2025Updated 2 months ago
- Generalized sparse regression for continuous and discrete data☆12Dec 12, 2025Updated 2 months ago
- JAX-accelerated nuclear equation of state inference and TOV solvers☆10Feb 7, 2026Updated last week
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆39Jan 16, 2026Updated 3 weeks ago
- Discovering Differential Equations with Physics-Informed Neural Networks and Symbolic Regression☆11Jul 28, 2023Updated 2 years ago
- Physics Informed Sparse Identification of Nonlinear Dynamics☆12Jan 3, 2025Updated last year
- A small library of paramaterizations and parameter constraints for PyTrees.☆50Jan 23, 2026Updated 3 weeks ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆53Updated this week
- Differentiable Likelihood for CMB Analysis☆25Jan 21, 2026Updated 3 weeks ago
- JAX reimplementation and bindings for the NVIDIA cuDecomp library☆52Updated this week
- All-KAN: All Kolmogorov-Arnold Networks We Know☆30Jan 9, 2026Updated last month
- An automatic differentiation system for dense and sparse problems☆13Jan 16, 2025Updated last year
- JAX code for modeling nearly-Keplerian orbits☆17Jan 8, 2026Updated last month
- Tackling the Curse of Dimensionality with Physics-Informed Neural Networks☆15May 17, 2024Updated last year
- Fast low-rank updates (LRU) of matrix determinants and pfaffians in JAX☆18Feb 6, 2026Updated last week
- 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 last month
- ☆403Nov 14, 2025Updated 3 months ago
- Unitful Quantities in JAX☆52Updated this week
- Julia implementation of the Kolmogorov-Arnold network with custom gradients for fast training.☆83Feb 7, 2026Updated last week
- A code for velocity-based Lagrangian and Eulerian PT expansions of redshift-space distortions.☆24Jul 11, 2025Updated 7 months ago
- Introductory workshop on using deep neural networks for symbolic regression☆15Updated this week
- Precise symbolic emulators of the linear and nonlinear matter power spectrum☆14Oct 22, 2025Updated 3 months ago
- Normalizing-flow enhanced sampling package for probabilistic inference in Jax☆255Oct 28, 2025Updated 3 months ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆510Updated this week
- [NeurIPS 2023 Spotlight] Separable Physics-Informed Neural Networks☆88Oct 25, 2023Updated 2 years ago
- Physics Informed Neural Networks (PINNs) + SPINNs + HyperPINNs + Adaptative Loss Weights with JAX 📓 Check out our various notebooks to g…☆45Updated this week
- ☆216Feb 1, 2026Updated last week
- ☆265Updated this week
- JAX-powered Cosmological Particle-Mesh N-body Solver☆58Dec 15, 2025Updated last month
- Stochastic Gradient MCMC for Jax☆18May 19, 2025Updated 8 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆13Jun 12, 2024Updated last year