Alternatives and similar repositories for jax_fdm

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

• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆103Updated last year
• tumaer / jax-sph
  JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework
  ☆75Updated 2 weeks ago
• GaoyuanWu / JaxSSO
  A differentiable finite element analysis solver for structural optimization based on JAX
  ☆34Updated 3 months ago
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆58Updated 4 years ago
• PrincetonLIPS / neural_fdm
  Real-time design of architectural structures with differentiable mechanics and neural networks [ICLR 2025]
  ☆22Updated 4 months ago
• tum-pbs / Solver-in-the-Loop
  Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers
  ☆166Updated 3 years ago
• JAX-DIPS / JAX-DIPS
  JAX-DIPS is a differentiable interfacial PDE solver.
  ☆46Updated last year
• barkm / torch-fenics
  PyTorch-FEniCS interface
  ☆102Updated 4 years ago
• UW-ERSL / AuTO
  ☆34Updated 2 years ago
• google-research / neural-structural-optimization
  Neural reparameterization improves structural optimization
  ☆125Updated 4 years ago
• tumaer / lagrangebench
  LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite
  ☆67Updated 9 months ago
• fireshape / fireshape
  A shape optimisation toolbox built on the Firedrake finite element library.
  ☆28Updated 2 months ago
• tum-pbs / PICT
  Official repository of the PICT Solver
  ☆36Updated last week
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆132Updated last year
• Autodesk / XLB
  XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML
  ☆408Updated 2 weeks ago
• mfschubert / sparsejac
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆65Updated 3 weeks ago
• hirani / pydec
  PyDEC: A Python Library for Discretizations of Exterior Calculus for simplicial complexes of any dimension embedded or not and for cubica…
  ☆96Updated 4 months ago
• dolfin-adjoint / pyadjoint
  The algorithmic differentation tool pyadjoint and add-ons.
  ☆106Updated last week
• martenlienen / finite-element-networks
  Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…
  ☆69Updated last year
• zfergus / topopt
  A Python Library for Topology Optimization
  ☆111Updated 3 years ago
• sigma-py / orthopy
  Orthogonal polynomials in all shapes and sizes.
  ☆187Updated last year
• pasteurlabs / tesseract-core
  Universal, autodiff-native software components for Simulation Intelligence. 📦
  ☆73Updated this week
• karlotness / nn-benchmark
  An extensible benchmark suite to evaluate data-driven physical simulation
  ☆29Updated 3 years ago
• loliverhennigh / PhantomGaze
  Simple GPU rendering of scientific data with Pytorch, Jax, CuPy, and Warp backends.
  ☆26Updated last year
• Naruki-Ichihara / feax
  A compact, high-performance finite element analysis engine built on JAX.
  ☆56Updated this week
• tum-pbs / diffSPH
  ☆39Updated 2 weeks ago
• holl- / PDE-Control
  Code for the ICLR 2020 paper "Learning to Control PDEs"
  ☆35Updated 5 years ago
• Ceyron / exponax
  Efficient Differentiable n-d PDE solvers in JAX.
  ☆47Updated last week
• tum-pbs / differentiable-piso
  Code repository for "Learned Turbulence Modelling with Differentiable Fluid Solvers"
  ☆39Updated 3 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆58Updated 3 months ago