Differentiable Finite Element Method with JAX

Microstructure Characterization and Reconstruction in Python

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs.kidger.site/diffrax/

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

A package for computing data-driven approximations to the Koopman operator.

Next generation FEniCS problem solving environment for solving finite element problems