Differentiable Finite Element Method with JAX
☆705Jun 13, 2026Updated 2 weeks ago
Alternatives and similar repositories for jax-fem
Users that are interested in jax-fem are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Additive manufacturing simulation with JAX.☆359Jul 11, 2025Updated 11 months ago
- Efficient GPU-computing simulation for differentiable crystal plasticity finite element method☆80Jun 5, 2026Updated 3 weeks ago
- GPU accelerated differentiable finite elements for solid mechanics with PyTorch☆115Updated this week
- ☆34Sep 21, 2022Updated 3 years ago
- GO-MELT: GPU-Optimized Multilevel Execution of LPBF Thermal simulations☆35Mar 24, 2026Updated 3 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Computational Fluid Dynamics in JAX☆948Feb 24, 2026Updated 4 months ago
- XLB: Accelerated Lattice Boltzmann (XLB) for Physics-based ML☆485May 29, 2026Updated last month
- A differentiable finite element analysis solver for structural optimization based on JAX☆41Dec 22, 2025Updated 6 months ago
- JAX-SPH: A Differentiable Smoothed Particle Hydrodynamics Framework☆80Oct 29, 2025Updated 8 months ago
- Simple finite element assemblers☆626Updated this week
- Differentiable Fluid Dynamics Package☆596Jun 18, 2026Updated last week
- ☆36Jul 19, 2023Updated 2 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆45Apr 15, 2026Updated 2 months ago
- Material Definition with Automatic Differentiation☆32Aug 28, 2025Updated 10 months ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Efficient Differentiable n-d PDE Solvers in JAX.☆209Jun 1, 2026Updated 3 weeks ago
- ☆35Jun 21, 2026Updated last week
- Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax☆557Jun 21, 2026Updated last week
- FEniCSx-based topology optimization supporting parallel computing☆96Nov 1, 2024Updated last year
- Next generation FEniCS problem solving environment☆1,153Updated this week
- An optimization tool for additive manufacturing using automatic differentiation☆61Jun 12, 2021Updated 5 years ago
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆109Nov 12, 2023Updated 2 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆134May 7, 2026Updated last month
- Jax wrapper for autograd-differentiable functions☆13Oct 20, 2025Updated 8 months ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- pjz is JAX and fdtd-z -- scalable photonic simulation and optimization☆21Mar 27, 2024Updated 2 years 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…☆32Apr 12, 2026Updated 2 months ago
- JAX-DIPS is a differentiable interfacial PDE solver.☆53Sep 14, 2024Updated last year
- ☆17Mar 28, 2024Updated 2 years ago
- Interpolation and function approximation with JAX☆275Jun 22, 2026Updated last week
- BONNI: Bayesian Optimization via Neural Network surrogates and Interior Point Optimization☆34Jun 19, 2026Updated last week
- Extension of DOLFINx implementing the concept of external operator☆46Updated this week
- Python wrapper for MFEM☆297Jun 22, 2026Updated last week
- IGAFrac (short for Isogeometric Analysis of Fracture) is a MATLAB code for fracture simulation.☆31May 10, 2022Updated 4 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- A differentiable PDE solving framework for machine learning☆1,897Updated this week
- Extension for dolfinx to handle multi-point constraints.☆55Jun 22, 2026Updated last week
- Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/☆2,060Jun 21, 2026Updated last week
- A Riks arclength solver implemented in FEniCS☆18Jan 8, 2026Updated 5 months ago
- measure minimum solid/void lengthscales in binary image☆22May 1, 2026Updated last month
- FEniCSx finite element basis evaluation library☆140Jun 18, 2026Updated last week
- Efficient forward- and reverse-mode sparse Jacobians using Jax☆70Dec 1, 2025Updated 6 months ago