numpy>=1.24
zarr>=2.16

[baseline]
torch>=2
zarr>=2.16
gcsfs>=2024.0

[cluster]
jax>=0.4
jax-md>=0.2
zarr>=2.16
gcsfs>=2024.0
fsspec>=2024.0

[dev]
pytest>=8

[sim]
jax>=0.4
jax-md>=0.2

[tpu]
jax[tpu]>=0.4
zarr>=2.16
gcsfs>=2024.0
fsspec>=2024.0
