Alternatives and similar repositories for exponax:

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

• Ceyron / pdequinox
  Neural Emulator Architectures in JAX.
  ☆16Updated 3 months 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…
  ☆59Updated 3 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆41Updated 3 months ago
• astanziola / fourier-neural-operator-flax
  An unofficial implementation of the Fourier Neural Operator in Flax
  ☆17Updated 8 months ago
• pnkraemer / tornadox
  Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…
  ☆20Updated 7 months ago
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆52Updated 3 years ago
• ASEM000 / finitediffX
  Finite difference tools in JAX
  ☆17Updated last year
• dmusekamp / al4pde
  AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers
  ☆13Updated 4 months ago
• VLSF / SNO
  Spectral Neural Operator
  ☆75Updated last year
• tum-pbs / unrolling
  ☆15Updated last month
• ASK-Berkeley / Neural-Spectral-Methods
  [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.
  ☆36Updated last year
• qiauil / ConvDO
  Convolutional Differential Operators for Physics-based Deep Learning Study
  ☆24Updated 6 months ago
• srigas / jaxKAN
  Kolmogorov-Arnold Networks built on JAX
  ☆56Updated 3 weeks ago
• brandstetter-johannes / LPSDA
  Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
  ☆49Updated last year
• ASEM000 / kernex
  Stencil computations in JAX
  ☆70Updated last year
• MagriLab / KolSol
  Pseudospectral Kolmogorov Flow Solver
  ☆37Updated last year
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆124Updated 5 months ago
• marvinpfoertner / linpde-gp
  Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"
  ☆25Updated 9 months ago
• Zhengyu-Huang / Operator-Learning
  ☆37Updated 2 years ago
• f0uriest / quadax
  Numerical quadrature with JAX
  ☆53Updated this week
• f0uriest / orthax
  Orthogonal polynomials with JAX
  ☆15Updated this week
• PredictiveIntelligenceLab / TRIPODS_Winter_School_2022
  Practicum on Supervised Learning in Function Spaces
  ☆32Updated 3 years ago
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆114Updated last month
• DifferentiableUniverseInitiative / jaxDecomp
  JAX bindings for the NVIDIA cuDecomp library
  ☆31Updated last week
• nickmcgreivy / WeakBaselinesMLPDE
  ☆22Updated 7 months ago
• yifanc96 / GaussianProcessPDEs
  An integrated demo: Gaussian processes for PDEs and inverse problems
  ☆13Updated last month
• IdahoLabResearch / BIhNNs
  The code enables to perform Bayesian inference in an efficient manner through the use of Hamiltonian Neural Networks (HNNs), Deep Neural …
  ☆14Updated 2 years ago
• RehMoritz / vmc_pde
  Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…
  ☆9Updated 2 years ago
• ScientificComputingCWI / SemesterProgramme-SciML
  Material for workshop and autumn school on scientific machine learning 2023
  ☆19Updated last year
• tum-pbs / PhiML
  Intuitive scientific computing with dimension types for Jax, PyTorch, TensorFlow & NumPy
  ☆82Updated this week