Alternatives and similar repositories for jax-fenics-adjoint

Users that are interested in jax-fenics-adjoint are comparing it to the libraries listed below

• IvanYashchuk / jax-fenics

  View on GitHub
  Differentiable interface to FEniCS for JAX
  ☆58May 30, 2021Updated 4 years ago
• IvanYashchuk / jax-firedrake

  View on GitHub
  Differentiable interface to Firedrake for JAX
  ☆15Feb 28, 2021Updated 4 years ago
• CMSL-HKUST / AmorFEA

  View on GitHub
  ☆13May 28, 2021Updated 4 years ago
• mattjj / ode-diff-notes

  View on GitHub
  ☆12Oct 8, 2020Updated 5 years ago
• DiffEqML / tutorials

  View on GitHub
  ☆11Apr 14, 2022Updated 3 years ago
• mlysy / rodeo-legacy

  View on GitHub
  Probabilistic Solution of Differential Equations
  ☆13Jun 19, 2022Updated 3 years ago
• flaport / klujax

  View on GitHub
  Solve sparse linear systems in JAX using the KLU algorithm
  ☆38Updated this week
• UW-ERSL / AuTO

  View on GitHub
  ☆35Jul 19, 2023Updated 2 years ago
• dolfin-adjoint / pyadjoint

  View on GitHub
  The algorithmic differentation tool pyadjoint and add-ons.
  ☆117Feb 9, 2026Updated last week
• mfschubert / sparsejac

  View on GitHub
  Efficient forward- and reverse-mode sparse Jacobians using Jax
  ☆69Dec 1, 2025Updated 2 months ago
• barkm / torch-fenics

  View on GitHub
  PyTorch-FEniCS interface
  ☆106Mar 29, 2021Updated 4 years ago
• juliagusak / neural-ode-metasolver

  View on GitHub
  Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561
  ☆25Mar 30, 2021Updated 4 years ago
• jxfzwar / topopt2d_MMA

  View on GitHub
  A python code for 2d topology optimization using MMA optimizers in NLOPT
  ☆12Feb 9, 2017Updated 9 years ago
• invrs-io / agjax

  View on GitHub
  Jax wrapper for autograd-differentiable functions
  ☆13Oct 20, 2025Updated 3 months ago
• deepmodeling / jax-fem

  View on GitHub
  Differentiable Finite Element Method with JAX
  ☆582Feb 8, 2026Updated last week
• francoismaze / topodiff

  View on GitHub
  Code, technical details and links to datasets and trained models for TopoDiff
  ☆56Nov 7, 2023Updated 2 years ago
• CMSL-HKUST / jax-am

  View on GitHub
  Additive manufacturing simulation with JAX.
  ☆340Jul 11, 2025Updated 7 months ago
• google / jax-cfd

  View on GitHub
  Computational Fluid Dynamics in JAX
  ☆920Jan 14, 2026Updated last month
• arpastrana / jax_fdm

  View on GitHub
  Auto-differentiable and hardware-accelerated force density method
  ☆95Jan 6, 2026Updated last month
• patrick-kidger / FasterNeuralDiffEq

  View on GitHub
  Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)
  ☆91Oct 19, 2022Updated 3 years ago
• googleinterns / invobs-data-assimilation

  View on GitHub
  ☆38May 20, 2021Updated 4 years ago
• FEniCS / basix

  View on GitHub
  FEniCSx finite element basis evaluation library
  ☆132Jan 26, 2026Updated 3 weeks ago
• martenlienen / finite-element-networks

  View on GitHub
  Reference implementation of Finite Element Networks as proposed in "Learning the Dynamics of Physical Systems from Sparse Observations wi…
  ☆69May 22, 2024Updated last year
• Daulbaev / IRDM

  View on GitHub
  ☆11Dec 6, 2020Updated 5 years ago
• pyroteus / pyroteus

  View on GitHub
  Goal-oriented error estimation and mesh adaptation for finite element problems solved using Firedrake
  ☆12Dec 5, 2023Updated 2 years ago
• LSDOlab / atomics

  View on GitHub
  ☆14Jul 23, 2022Updated 3 years ago
• ucl-bug / jaxdf

  View on GitHub
  A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
  ☆132Feb 9, 2026Updated last week
• tristandeleu / jax-comln

  View on GitHub
  Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)
  ☆25Mar 4, 2022Updated 3 years ago
• google / jaxopt

  View on GitHub
  Hardware accelerated, batchable and differentiable optimizers in JAX.
  ☆1,026Dec 17, 2025Updated last month
• microsoft / knossos-ksc

  View on GitHub
  Compiler with automatic differentiation
  ☆49Oct 18, 2023Updated 2 years ago
• dfm / extending-jax

  View on GitHub
  Extending JAX with custom C++ and CUDA code
  ☆403Aug 18, 2024Updated last year
• analysiscenter / pydens

  View on GitHub
  PyDEns is a framework for solving Ordinary and Partial Differential Equations (ODEs & PDEs) using neural networks
  ☆313Feb 9, 2024Updated 2 years ago
• neo-fetch / nvidia-modulus-airfoil-optimisation

  View on GitHub
  Using NVIDIA modulus for airfoil optimizations at different angles.
  ☆25Apr 5, 2023Updated 2 years ago
• greydanus / optimize_wing

  View on GitHub
  We simulate a wind tunnel, place a rectangular occlusion in it, and then use gradient descent to turn the occlusion into a wing.
  ☆27Oct 16, 2020Updated 5 years ago
• PittAMRL / StressTopOpt

  View on GitHub
  Code sample for "An efficient 146-line 3D sensitivity analysis code of stress-based topology optimization written in MATLAB"
  ☆33Aug 5, 2024Updated last year
• patrick-kidger / diffrax

  View on GitHub
  Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/
  ☆1,898Feb 8, 2026Updated last week
• patrick-kidger / lineax

  View on GitHub
  Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
  ☆510Updated this week
• ermongroup / fast_feedforward_computation

  View on GitHub
  Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021
  ☆29Sep 25, 2021Updated 4 years ago
• benvial / pytheas

  View on GitHub
  Python Electromagnetic Analysis and Simulation with the Finite Element Method
  ☆15Mar 16, 2021Updated 4 years ago