IvanYashchuk / jax-fenicsView external linksLinks
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
- ☆15Nov 20, 2023Updated 2 years ago
- Auto-differentiable and hardware-accelerated force density method☆95Jan 6, 2026Updated last month
- FEniCSx finite element basis evaluation library☆132Jan 26, 2026Updated 3 weeks ago
- ☆15Apr 20, 2023Updated 2 years ago
- Differentiable interface to Firedrake for JAX☆15Feb 28, 2021Updated 4 years ago
- The algorithmic differentation tool pyadjoint and add-ons.☆117Feb 9, 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☆138Feb 9, 2026Updated last week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆513Updated this week
- Stencil computations in JAX☆70Nov 15, 2025Updated 3 months ago
- Turning SymPy expressions into JAX functions☆45Mar 17, 2021Updated 4 years ago
- Extension for dolfinx to handle multi-point constraints.☆51Updated this week
- Additive manufacturing simulation with JAX.☆340Jul 11, 2025Updated 7 months ago
- Expreriments with Nitsche method of setting boundary conditions in FEniCS☆22Jun 16, 2014Updated 11 years ago
- A Python library for isogeometric analysis (IGA) using FEniCS.☆113Sep 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☆38Updated this week
- PyTorch-FEniCS interface☆106Mar 29, 2021Updated 4 years ago
- ☆35Jul 19, 2023Updated 2 years ago
- FEniCS on GPU takes advantage of CUDA cores to solve SPARSE matrix using cuPy and SciPy libraries.☆20Jul 4, 2021Updated 4 years ago
- Computational Fluid Dynamics in JAX☆920Jan 14, 2026Updated last month
- A flexible framework for solving PDEs with modern spectral methods.☆656Jan 21, 2026Updated 3 weeks ago
- SymPy based framework for optimized code generation for BSSN formulation of Einstein equation for heterogeneous platforms.☆11Aug 18, 2025Updated 5 months ago
- Python for IsoGeometric Analysis☆10Nov 22, 2016Updated 9 years ago
- 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
- Exponential families for JAX☆76Feb 1, 2026Updated 2 weeks ago
- Simple finite element assemblers☆608Jan 30, 2026Updated 2 weeks ago
- PFASST++ is a C++ implementation of the "parallel full approximation scheme in space and time" (PFASST) algorithm☆37Dec 12, 2016Updated 9 years ago
- Reparameterizing Engineering Designs for Augmented Multi-objective Optimization☆10Feb 18, 2023Updated 2 years ago
- Neural operator learning of heterogeneous mechanobiological insults contributing to aortic aneurysms☆12Nov 4, 2024Updated last year
- SParse lINear solvers in JAX☆26Feb 2, 2026Updated last week
- FEniCSx Fracture mechanics examples☆35Dec 29, 2025Updated last month
- Forward mode laplacian implemented in JAX tracer☆30Jan 7, 2026Updated last month
- Spectral Neural Operator☆79Dec 20, 2023Updated 2 years ago
- Differentiable Finite Element Method with JAX☆582Feb 8, 2026Updated last week
- Benchmark for learning stiff problems using physics-informed machine learning☆12Dec 15, 2021Updated 4 years ago
- A Riks arclength solver implemented in FEniCS☆13Jan 8, 2026Updated last month
- Neural Network Based Lattice Boltzmann solver☆26Oct 25, 2018Updated 7 years ago