jax>=0.8
jaxlib!=0.10.2,>=0.8

[cpu]

[cpu:sys_platform == "linux" and platform_machine == "x86_64"]
lammps-jax-pjrt-cpu>=0.11.2

[dist]
ase>=3.22
e3nn-jax>=0.20
jax!=0.11.2
jax-md>=0.2.27
nequix>=0.4

[test]
pytest>=7
numpy>=1.24
