numpy>=1.21.0

[all]
mlx>=0.0.7
torch>=2.0.0
jax>=0.4.0

[all:sys_platform == "darwin"]
jax-metal>=0.1.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=22.0.0
isort>=5.10.0
flake8>=4.0.0

[jax]
jax>=0.4.0

[jax:sys_platform == "darwin"]
jax-metal>=0.1.0

[mlx]
mlx>=0.0.7

[torch]
torch>=2.0.0
