jax<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
shapely>=2.0
paramax>=0.0.5
blackjax>=1.2
fenics-basix
matfree<0.7,>=0.6.1

[amg]
jaxamg[cuda13]<0.2,>=0.1.3

[cuda]
jax[cuda]<0.11,>=0.10.1

[cudss]

[cudss:platform_machine == "x86_64" and sys_platform == "linux"]
nvmath-python>=1.0
nvidia-cudss-cu12
cupy-cuda12x

[dev]
pytest>=7.0
black>=22.0
flake8>=4.0
mypy>=0.950
paramax
flax
xprof
nevergrad
pyfiglet

[fem]
jax-numerical-operators[cudss,mesh,pardiso]

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

[mesh]
mmgpy

[pardiso]

[pardiso:platform_machine == "x86_64"]
pypardiso>=0.4.4

[rcwa]
fmmax
