numpy>=1.24

[accelerators]
jax[cuda12]>=0.4.30
jax[tpu]>=0.4.30
mlx>=0.18

[cuda]
jax[cuda12]>=0.4.30

[dev]
pytest>=7.0

[jax]
jax>=0.4.30

[mps]
mlx>=0.18

[tpu]
jax[tpu]>=0.4.30
