numpy>=1.12
jax>=0.3.16
matplotlib
msgpack
optax
rich~=11.1
typing_extensions>=4.1.1
PyYAML>=5.4.1

[testing]
atari-py==0.2.5
clu
gym==0.18.3
jaxlib
jraph>=0.0.6dev0
ml-collections
opencv-python
pytest
pytest-cov
pytest-custom_exit_code
pytest-xdist==1.34.0
pytype
sentencepiece
svn
tensorflow_text>=2.4.0
tensorflow_datasets
tensorflow
torch
