GitPython
tensorstore>=0.1.14
file-or-name
six
scipy
numba
msgpack

[:python_version < "3.10.0"]
importlib_metadata

[:python_version < "3.8.0"]
typing_extensions

[:python_version < "3.9.0"]
importlib_resources

[all]
flax
tensorflow
torch

[docs]
sphinx
numpydoc

[flax]
flax

[pytorch]
torch

[tensorflow]
tensorflow

[test]
pytest
