
[all]
torch
numpy
tensorflow
jax
flax
black==22.3
isort>=5.5.4
flake8>=3.8.3
click==8.0.4
setuptools_rust
huggingface_hub
pytest
pytest-benchmark
h5py

[dev]
torch
numpy
tensorflow
jax
flax
black==22.3
isort>=5.5.4
flake8>=3.8.3
click==8.0.4
setuptools_rust
huggingface_hub
pytest
pytest-benchmark
h5py

[jax]
jax
flax

[numpy]
numpy

[quality]
black==22.3
isort>=5.5.4
flake8>=3.8.3
click==8.0.4

[tensorflow]
tensorflow

[testing]
setuptools_rust
huggingface_hub
pytest
pytest-benchmark
h5py
numpy

[torch]
torch
