numpy>=1.20.0

[bfloat16]
ml_dtypes>=0.3.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0

[torch]
torch>=2.0.0
