deepmodeling / jax-femView external linksLinks
Differentiable Finite Element Method with JAX
☆582Feb 8, 2026Updated last week
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.☆340Jul 11, 2025Updated 7 months ago
- Efficient GPU-computing simulation for differentiable crystal plasticity finite element method☆72Jan 22, 2026Updated 3 weeks ago
- GPU accelerated differentiable finite elements for solid mechanics with PyTorch☆89Feb 5, 2026Updated last week
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆440Jan 20, 2026Updated 3 weeks ago
- Computational Fluid Dynamics in JAX☆920Jan 14, 2026Updated last month
- Simple finite element assemblers☆608Jan 30, 2026Updated 2 weeks ago
- Differentiable Fluid Dynamics Package☆510Jan 29, 2026Updated 2 weeks ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆77Oct 29, 2025Updated 3 months ago
- A differentiable finite element analysis solver for structural optimization based on JAX☆40Dec 22, 2025Updated last month
- Solve sparse linear systems in JAX using the KLU algorithm☆38Updated this week
- Efficient Differentiable n-d PDE Solvers in JAX.☆53Updated this week
- ☆32Sep 21, 2022Updated 3 years ago
- ☆32Jan 2, 2026Updated last month
- GO-MELT: GPU-Optimized Multilevel Execution of LPBF Thermal simulations☆32Dec 3, 2025Updated 2 months ago
- Material Definition with Automatic Differentiation☆32Aug 28, 2025Updated 5 months ago
- Next generation FEniCS problem solving environment☆1,048Updated this week
- ☆35Jul 19, 2023Updated 2 years ago
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆510Updated this week
- 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☆287Jan 15, 2026Updated last month
- Jax wrapper for autograd-differentiable functions☆13Oct 20, 2025Updated 3 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…☆26Nov 20, 2025Updated 2 months ago
- FEniCSx-based topology optimization supporting parallel computing☆82Nov 1, 2024Updated last year
- JAX-DIPS is a differentiable interfacial PDE solver.☆49Sep 14, 2024Updated last year
- A differentiable PDE solving framework for machine learning☆1,806Feb 6, 2026Updated last week
- An optimization tool for additive manufacturing using automatic differentiation☆59Jun 12, 2021Updated 4 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆132Feb 9, 2026Updated last week
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆69Dec 1, 2025Updated 2 months ago
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆1,898Feb 8, 2026Updated last week
- Scikit for Gmsh to generate 3D finite element mesh☆73Updated this week
- A Riks arclength solver implemented in FEniCS☆13Jan 8, 2026Updated last month
- Interpolation and function approximation with JAX☆248Feb 2, 2026Updated last week
- Differentiable interface to FEniCS for JAX☆58May 30, 2021Updated 4 years ago
- finite element analysis for continuum mechanics of solid bodies☆163Updated this week
- Lightweight, general, scalable C++ library for finite element methods☆2,096Updated this week
- Extension for dolfinx to handle multi-point constraints.☆51Updated this week
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆541Updated this week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆513Updated this week