Differentiable interface to FEniCS for JAX
☆58May 30, 2021Updated 4 years ago
Alternatives and similar repositories for jax-fenics
Users that are interested in jax-fenics are comparing it to the libraries listed below
Sorting:
- Differentiable interface to FEniCS/Firedrake for JAX using dolfin-adjoint/pyadjoint☆107Nov 12, 2023Updated 2 years ago
- Auto-differentiable and hardware-accelerated force density method☆96Mar 1, 2026Updated last week
- ☆15Nov 20, 2023Updated 2 years ago
- FEniCSx finite element basis evaluation library☆132Updated this week
- ☆15Apr 20, 2023Updated 2 years ago
- Differentiable interface to Firedrake for JAX☆15Feb 28, 2021Updated 5 years ago
- The algorithmic differentation tool pyadjoint and add-ons.☆118Feb 27, 2026Updated last week
- Source code for deep learning-based reduced order models in cardiac electrophysiology. Available on doi.org/10.1371/journal.pone.0239416.☆16Sep 7, 2023Updated 2 years ago
- UFL - Unified Form Language☆139Updated this week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆517Updated this week
- Stencil computations in JAX☆71Nov 15, 2025Updated 3 months ago
- Extension for dolfinx to handle multi-point constraints.☆52Feb 25, 2026Updated last week
- Turning SymPy expressions into JAX functions☆45Mar 17, 2021Updated 4 years ago
- Additive manufacturing simulation with JAX.☆339Jul 11, 2025Updated 7 months ago
- ☆21Jun 28, 2025Updated 8 months ago
- Spectral Neural Operator☆79Dec 20, 2023Updated 2 years ago
- Expreriments with Nitsche method of setting boundary conditions in FEniCS☆22Jun 16, 2014Updated 11 years ago
- Computational Fluid Dynamics in JAX☆927Feb 24, 2026Updated last week
- A Python library for isogeometric analysis (IGA) using FEniCS.☆114Sep 14, 2024Updated last year
- Solving wave problems using FEniCS-X☆23Oct 21, 2021Updated 4 years ago
- Solve sparse linear systems in JAX using the KLU algorithm☆38Feb 19, 2026Updated 2 weeks ago
- PyTorch-FEniCS interface☆107Mar 29, 2021Updated 4 years ago
- A flexible framework for solving PDEs with modern spectral methods.☆662Updated this week
- Utility functions for the automatic generation of nonlinear DG FEM formulations using UFL☆22Nov 14, 2025Updated 3 months ago
- FEniCS on GPU takes advantage of CUDA cores to solve SPARSE matrix using cuPy and SciPy libraries.☆20Jul 4, 2021Updated 4 years ago
- Simple walkthrough tutorial for getting started with petsc4py☆41Sep 13, 2011Updated 14 years ago
- Simple finite element assemblers☆607Jan 30, 2026Updated last month
- Code for the paper "Variational Monte Carlo Approach to Partial Differential Equations with Neural Networks" (https://arxiv.org/abs/2206.…☆10Aug 8, 2022Updated 3 years ago
- SymPy based framework for optimized code generation for BSSN formulation of Einstein equation for heterogeneous platforms.☆11Aug 18, 2025Updated 6 months ago
- Python for IsoGeometric Analysis☆10Nov 22, 2016Updated 9 years ago
- ☆12Aug 2, 2025Updated 7 months ago
- Exponential families for JAX☆76Updated this week
- PFASST++ is a C++ implementation of the "parallel full approximation scheme in space and time" (PFASST) algorithm☆37Dec 12, 2016Updated 9 years ago
- SParse lINear solvers in JAX☆30Feb 22, 2026Updated 2 weeks ago
- Reparameterizing Engineering Designs for Augmented Multi-objective Optimization☆10Feb 18, 2023Updated 3 years ago
- Neural operator learning of heterogeneous mechanobiological insults contributing to aortic aneurysms☆12Nov 4, 2024Updated last year
- FEniCSx Fracture mechanics examples☆35Dec 29, 2025Updated 2 months ago
- Forward mode laplacian implemented in JAX tracer☆30Jan 7, 2026Updated 2 months ago
- Benchmark for learning stiff problems using physics-informed machine learning☆12Dec 15, 2021Updated 4 years ago