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
  ☆101Updated last year
• tumaer / jax-sph
  JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework
  ☆75Updated 8 months ago
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆58Updated 4 years ago
• GaoyuanWu / JaxSSO
  A differentiable finite element analysis solver for structural optimization based on JAX
  ☆32Updated 2 months ago
• PrincetonLIPS / neural_fdm
  Real-time design of architectural structures with differentiable mechanics and neural networks [ICLR 2025]
  ☆21Updated 3 months ago
• JAX-DIPS / JAX-DIPS
  JAX-DIPS is a differentiable interfacial PDE solver.
  ☆46Updated last year
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆129Updated last year
• UW-ERSL / AuTO
  ☆34Updated 2 years ago
• barkm / torch-fenics
  PyTorch-FEniCS interface
  ☆102Updated 4 years ago
• google-research / neural-structural-optimization
  Neural reparameterization improves structural optimization
  ☆125Updated 4 years ago
• tum-pbs / Solver-in-the-Loop
  Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers
  ☆163Updated 2 years ago
• mfschubert / sparsejac
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆64Updated 2 weeks ago
• tum-pbs / PICT
  Official repository of the PICT Solver
  ☆34Updated 3 months ago
• fireshape / fireshape
  A shape optimisation toolbox built on the Firedrake finite element library.
  ☆28Updated last month
• 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
• Autodesk / XLB
  XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML
  ☆389Updated last month
• tumaer / lagrangebench
  LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite
  ☆65Updated 8 months ago
• tum-pbs / diffSPH
  ☆37Updated 3 weeks ago
• Naruki-Ichihara / feax
  A compact, high-performance finite element analysis engine built on JAX.
  ☆53Updated 2 weeks ago
• karlotness / nn-benchmark
  An extensible benchmark suite to evaluate data-driven physical simulation
  ☆29Updated 3 years ago
• dolfin-adjoint / pyadjoint
  The algorithmic differentation tool pyadjoint and add-ons.
  ☆104Updated last week
• hirani / pydec
  PyDEC: A Python Library for Discretizations of Exterior Calculus for simplicial complexes of any dimension embedded or not and for cubica…
  ☆96Updated 2 months ago
• Ceyron / exponax
  Efficient Differentiable n-d PDE solvers in JAX.
  ☆47Updated 10 months ago
• loliverhennigh / PhantomGaze
  Simple GPU rendering of scientific data with Pytorch, Jax, CuPy, and Warp backends.
  ☆26Updated last year
• zfergus / topopt
  A Python Library for Topology Optimization
  ☆110Updated 3 years ago
• tum-pbs / dl-surrogates
  ☆32Updated 2 years ago
• sigma-py / orthopy
  Orthogonal polynomials in all shapes and sizes.
  ☆187Updated last year
• dl4to / dl4to
  DL4TO is a Python library for 3D topology optimization that is based on PyTorch and allows easy integration with neural networks.
  ☆94Updated 10 months ago
• ASK-Berkeley / Neural-Spectral-Methods
  [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.
  ☆46Updated last year
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆56Updated last month