Alternatives and similar repositories for jax_fdm:

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

• GaoyuanWu / JaxSSO
  A differentiable finite element analysis solver for structural optimization based on JAX
  ☆24Updated 4 months ago
• tumaer / jax-sph
  JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework
  ☆58Updated last month
• UW-ERSL / AuTO
  ☆34Updated last year
• IvanYashchuk / jax-fenics-adjoint
  Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint
  ☆93Updated last year
• IvanYashchuk / jax-fenics
  Differentiable interface to FEniCS for JAX
  ☆52Updated 3 years ago
• fireshape / fireshape
  A shape optimisation toolbox built on the Firedrake finite element library.
  ☆28Updated 2 weeks ago
• JAX-DIPS / JAX-DIPS
  JAX-DIPS is a differentiable interfacial PDE solver.
  ☆41Updated 5 months ago
• Autodesk / XLB
  XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML
  ☆261Updated this week
• barkm / torch-fenics
  PyTorch-FEniCS interface
  ☆99Updated 3 years ago
• google-research / neural-structural-optimization
  Neural reparameterization improves structural optimization
  ☆118Updated 3 years ago
• mfschubert / sparsejac
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆55Updated 2 months ago
• ucl-bug / jaxdf
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆124Updated 5 months ago
• karlotness / nn-benchmark
  An extensible benchmark suite to evaluate data-driven physical simulation
  ☆25Updated 3 years ago
• tum-pbs / Solver-in-the-Loop
  Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solvers
  ☆152Updated 2 years ago
• ASK-Berkeley / Neural-Spectral-Methods
  [ICLR 2024] Neural Spectral Methods: Self-supervised learning in the spectral domain.
  ☆36Updated last year
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆45Updated 2 weeks ago
• martenlienen / finite-element-networks
  Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…
  ☆70Updated 8 months ago
• arpastrana / compas_cem
  Inverse design of 3D truss networks with automatic differentiation
  ☆38Updated 3 weeks ago
• gdmcbain / fenics-tuto-in-skfem
  reproducing the examples from the FEniCS tutorial in scikit-fem
  ☆15Updated 3 years ago
• zfergus / topopt
  A Python Library for Topology Optimization
  ☆105Updated 2 years ago
• tumaer / lagrangebench
  LagrangeBench: A Lagrangian Fluid Mechanics Benchmarking Suite
  ☆62Updated last month
• meshpro / meshplex
  Compute interesting points, areas, and volumes in simplex meshes of any dimension.
  ☆105Updated 8 months ago
• MilesCranmer / sympy2jax
  Turning SymPy expressions into JAX functions
  ☆44Updated 3 years ago
• sandialabs / optimism
  Computational solid mechanics made easy with Jax
  ☆41Updated this week
• zhehaoli1999 / taichi-TopOpt
  A topology optimization framework written in Taichi programming language.
  ☆61Updated 3 years ago
• sigma-py / orthopy
  Orthogonal polynomials in all shapes and sizes.
  ☆184Updated last year
• adtzlr / matadi
  Material Definition with Automatic Differentiation
  ☆22Updated last week
• dolfin-adjoint / pyadjoint
  The algorithmic differentation tool pyadjoint and add-ons.
  ☆93Updated this week
• tum-pbs / dl-surrogates
  ☆31Updated 2 years ago
• sblauth / cashocs
  computational adjoint-based shape optimization and optimal control software for python
  ☆56Updated this week