jax[cuda]<0.11,>=0.10.1
equinox>=0.11.0
optax>=0.2.0
orbax-checkpoint>=0.6.0
foundax>=0.2.0
pygmsh>=7.0.0
cloudpickle>=3.0.0
einops>=0.7.0
matplotlib>=3.8.0
pylotte>=0.3
shapely>=2.0
paramax>=0.0.5

[dev]
pytest>=7.0
black>=22.0
flake8>=4.0
mypy>=0.950
feax>=0.5.1
diffrax
optimistix
lineax
paramax
flax
xprof
pylotte
nevergrad
pyfiglet
fenics-basix

[fem]
feax>=0.5.1
diffrax
optimistix
lineax

[iree]
iree-base-compiler
iree-base-runtime
