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 6 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…
  ☆69Updated 5 months ago
• astanziola / fourier-neural-operator-flax
  An unofficial implementation of the Fourier Neural Operator in Flax
  ☆17Updated 11 months ago
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆53Updated 3 years ago
• dmusekamp / al4pde
  AL4PDE: A Benchmark for Active Learning for Neural PDE Solvers
  ☆22Updated 2 months ago
• tum-pbs / unrolling
  ☆17Updated 4 months ago
• VLSF / SNO
  Spectral Neural Operator
  ☆78Updated last year
• qiauil / ConvDO
  Convolutional Differential Operators for Physics-based Deep Learning Study
  ☆24Updated 9 months ago
• MagriLab / KolSol
  Pseudospectral Kolmogorov Flow Solver
  ☆38Updated last year
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆125Updated 7 months ago
• Zhengyu-Huang / Operator-Learning
  ☆41Updated 2 years 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…
  ☆21Updated 9 months ago
• pnkraemer / probdiffeq
  Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…
  ☆46Updated last month
• ASEM000 / finitediffX
  Finite difference tools in JAX
  ☆19Updated last year
• brandstetter-johannes / LPSDA
  Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"
  ☆49Updated last year
• ASK-Berkeley / Neural-Spectral-Methods
  [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.
  ☆41Updated last year
• pnkraemer / matfree
  Matrix-free linear algebra in JAX.
  ☆117Updated 3 months ago
• tomoleary / dino
  Derivative-Informed Neural Operator: An Efficient Framework for High-Dimensional Parametric Derivative Learning
  ☆17Updated last year
• amazon-science / probconserv
  Datasets and code for results presented in the ProbConserv paper
  ☆53Updated 10 months ago
• htlambley / functional_autoencoders
  Autoencoders in Function Space: Resolution-invariant autoencoders for functional data based on the functional variational autoencoder (FV…
  ☆13Updated 4 months ago
• PredictiveIntelligenceLab / NOMAD
  ☆16Updated 8 months ago
• mfschubert / sparsejac
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆55Updated 5 months ago
• srigas / jaxKAN
  Kolmogorov-Arnold Networks built on JAX
  ☆76Updated 3 weeks ago
• marvinpfoertner / linpde-gp
  Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"
  ☆25Updated last year
• tumaer / lagrangebench
  LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite
  ☆63Updated 4 months ago
• pnkraemer / probfindiff
  Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…
  ☆11Updated last year
• lbl-camera / fvGP
  A software package for flexible HPC GPs
  ☆16Updated this week
• yifanc96 / GaussianProcessPDEs
  An integrated demo: Gaussian processes for PDEs and inverse problems
  ☆14Updated 3 months ago
• dimitra-maoutsa / DeterministicParticleFlowControl
  Repository for Deterministic Particle Flow Control framework
  ☆10Updated 2 years ago
• NBoulle / RationalNets
  Code for the paper "Rational neural networks", NeurIPS 2020
  ☆28Updated 4 years ago