numpy
scipy
matplotlib
tqdm
torch>=2.9
jax>=0.6.2
orbax
jax-tqdm

[cuda12]

[cuda12:sys_platform == "linux" and platform_machine == "x86_64"]
jax[cuda12]>=0.6.2

[cuda13]

[cuda13:sys_platform == "linux" and platform_machine == "x86_64"]
jax[cuda13]>=0.6.2

[mesh]
meshio
gmsh
