numpy>=1.12
jax>=0.2.13
matplotlib
msgpack
optax

[:python_version < "3.7"]
dataclasses

[testing]
atari-py==0.2.5
clu
gym==0.18.3
jaxlib
ml-collections
opencv-python
pytest
pytest-cov
pytest-xdist==1.34.0
pytype==2021.5.25
sentencepiece
svn
tensorflow-cpu>=2.4.0
tensorflow_text>=2.4.0
tensorflow_datasets
tensorflow==2.4.1
