numpy>=1.22.0
matplotlib>=3.5.0
psutil>=5.9.0

[full]
jax>=0.4.0
jaxlib>=0.4.0
cupy-cuda12x>=12.0.0

[gpu]
cupy-cuda12x>=12.0.0

[jax]
jax>=0.4.0
jaxlib>=0.4.0
