absl-py
funcsigs
gin-config
gym
jax
jaxlib
matplotlib
numpy
psutil
scipy
six
tensorflow-datasets
tensorflow-text

[t2t]
tensor2tensor

[t5]
t5>=0.4.0

[tensorflow]
tensorflow>=1.15.0

[tensorflow_gpu]
tensorflow-gpu>=1.15.0

[tests]
attrs
jupyter
mock
parameterized
pylint
pytest
wrapt==1.11.*
