Alternatives and similar repositories for jaxKAN

Users that are interested in jaxKAN are comparing it to the libraries listed below

• Ceyron / exponax
  Efficient Differentiable n-d PDE solvers in JAX.
  ☆51Updated 2 weeks ago
• Ceyron / pdequinox
  Neural Emulator Architectures in JAX.
  ☆20Updated last year
• qiauil / torchfsm
  TorchFSM: Fourier Spectral Method with PyTorch
  ☆50Updated last week
• tum-pbs / PhiML
  Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy
  ☆93Updated last week
• f0uriest / interpax
  Interpolation and function approximation with JAX
  ☆226Updated last week
• danielward27 / flowjax
  ☆198Updated 2 weeks ago
• tum-pbs / apebench
  [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…
  ☆88Updated this week
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆148Updated 2 weeks ago
• astanziola / fourier-neural-operator-flax
  An unofficial implementation of the Fourier Neural Operator in Flax
  ☆18Updated last year
• VLSF / SNO
  Spectral Neural Operator
  ☆78Updated last year
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆132Updated last year
• PolymathicAI / multiple_physics_pretraining
  Code for paper "Multiple Physics Pretraining for Physical Surrogate Models
  ☆178Updated 11 months ago
• marvinpfoertner / linpde-gp
  Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"
  ☆30Updated last year
• benmoseley / intro-to-jax-workshop
  Introduction to JAX Workshop @ ETH Zurich, 25 June 2024
  ☆38Updated 7 months ago
• YaoGroup / MultistageNN
  Codes associated with the manuscript titled "Multi-stage neural networks: Function approximator of machine precision"
  ☆49Updated last year
• dmusekamp / al4pde
  AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers
  ☆29Updated 5 months ago
• pdearena / pdearena
  ☆274Updated last year
• rezaakb / pinns-jax
  PINNs-JAX, Physics-informed Neural Networks (PINNs) implemented in JAX.
  ☆55Updated last year
• dirmeier / sbijax
  Simulation-based inference in JAX
  ☆39Updated 7 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆53Updated this week
• patrick-kidger / quax
  Multiple dispatch over abstract array types in JAX.
  ☆136Updated this week
• SciML / FluxNeuralOperators.jl
  DeepONets, (Fourier) Neural Operators, Physics-Informed Neural Operators, and more in Julia
  ☆289Updated last year
• tum-pbs / ConFIG
  [ICLR2025 Spotlight] Official implementation of Conflict-Free Inverse Gradients Method
  ☆89Updated 5 months ago
• camlab-ethz / poseidon
  Code for the paper "Poseidon: Efficient Foundation Models for PDEs"
  ☆159Updated 7 months ago
• dodgebc / jaxkd
  Minimal JAX implementation of k-nearest neighbors using a k-d tree.
  ☆50Updated 4 months ago
• PredictiveIntelligenceLab / jaxpi
  ☆372Updated last month
• matteoguarrera / awesome-equinox
  Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox
  ☆16Updated last year
• neuraloperator / CoDA-NO
  Codomain attention neural operator for single to multi-physics PDE adaptation.
  ☆69Updated 5 months ago
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆72Updated last week
• alonfnt / bayex
  Minimal Implementation of Bayesian Optimization in JAX
  ☆100Updated 6 months ago