Differentiable Finite Element Method with JAX
☆601Feb 8, 2026Updated last month
Alternatives and similar repositories for jax-fem
Users that are interested in jax-fem are comparing it to the libraries listed below
Sorting:
- Additive manufacturing simulation with JAX.☆339Jul 11, 2025Updated 7 months ago
- Efficient GPU-computing simulation for differentiable crystal plasticity finite element method☆72Jan 22, 2026Updated last month
- GPU accelerated differentiable finite elements for solid mechanics with PyTorch☆93Updated this week
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆442Jan 20, 2026Updated last month
- Computational Fluid Dynamics in JAX☆927Feb 24, 2026Updated last week
- Simple finite element assemblers☆607Jan 30, 2026Updated last month
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆77Oct 29, 2025Updated 4 months ago
- Differentiable Fluid Dynamics Package☆536Updated this week
- A differentiable finite element analysis solver for structural optimization based on JAX☆40Dec 22, 2025Updated 2 months ago
- Solve sparse linear systems in JAX using the KLU algorithm☆38Feb 19, 2026Updated 2 weeks ago
- ☆33Updated this week
- GO-MELT: GPU-Optimized Multilevel Execution of LPBF Thermal simulations☆32Dec 3, 2025Updated 3 months ago
- Material Definition with Automatic Differentiation☆32Aug 28, 2025Updated 6 months ago
- ☆33Sep 21, 2022Updated 3 years ago
- Next generation FEniCS problem solving environment☆1,067Mar 3, 2026Updated last week
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆522Mar 1, 2026Updated last week
- ☆35Jul 19, 2023Updated 2 years ago
- pjz is JAX and fdtd-z -- scalable photonic simulation and optimization☆21Mar 27, 2024Updated last year
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆107Nov 12, 2023Updated 2 years ago
- Python wrapper for MFEM☆289Feb 22, 2026Updated 2 weeks ago
- Jax wrapper for autograd-differentiable functions☆13Oct 20, 2025Updated 4 months ago
- dolfinx_materials is a Python add-on package to the dolfinx interface to the FEniCSx project. It enables the user to define complex mater…☆26Feb 17, 2026Updated 3 weeks ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆155Mar 2, 2026Updated last week
- FEniCSx-based topology optimization supporting parallel computing☆82Nov 1, 2024Updated last year
- JAX-DIPS is a differentiable interfacial PDE solver.☆50Sep 14, 2024Updated last year
- A differentiable PDE solving framework for machine learning☆1,829Updated this week
- An optimization tool for additive manufacturing using automatic differentiation☆59Jun 12, 2021Updated 4 years ago
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆70Dec 1, 2025Updated 3 months ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆133Feb 20, 2026Updated 2 weeks ago
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,923Feb 23, 2026Updated 2 weeks ago
- A Riks arclength solver implemented in FEniCS☆14Jan 8, 2026Updated 2 months ago
- Scikit for Gmsh to generate 3D finite element mesh☆74Mar 2, 2026Updated last week
- Interpolation and function approximation with JAX☆252Mar 2, 2026Updated last week
- Differentiable interface to FEniCS for JAX☆59May 30, 2021Updated 4 years ago
- finite element analysis for continuum mechanics of solid bodies☆165Updated this week
- Lightweight, general, scalable C++ library for finite element methods☆2,116Updated this week
- Extension for dolfinx to handle multi-point constraints.☆52Mar 2, 2026Updated last week
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆546Feb 16, 2026Updated 3 weeks ago
- A Python finite element code☆334Feb 28, 2026Updated last week