numpy>=1.21.0
jax>=0.4.0
jaxlib>=0.4.0
flax>=0.7.0
optax>=0.1.0
scipy>=1.7.0
tokenizers>=0.13.0
datasets>=2.0.0

[:platform_python_implementation != "PyPy"]
blosc>=1.0.0

[dev]
pytest>=6.0
pytest-cov>=2.0
mypy>=0.910
black>=21.0
flake8>=3.8

[full]
torch>=1.9.0
tensorflow>=2.8.0
