safetensors
numpy

[dev]
pre-commit

[docs]
mkdocs-material
mkdocs-git-revision-date-localized-plugin
mkdocstrings[python]
pymdown-extensions

[test]
pytest
torch
torchvision
tensorflow
tensorflow-hooks
flax
transformers
