absl-py==1.0.0
numpy==1.21.6
pandas==1.3.5
protobuf==3.20.*
six==1.16.0
tensorflow==2.9.0
tensorflow_datasets==4.8.2
tensorflow_probability==0.17.0
tensorflow_addons==0.18.0
gputil==1.4.0
psutil==5.9.1
clu==0.0.7

[criteo1tb]
scikit-learn==1.0.1

[dev]
isort==5.10.1
pylint==2.16.1
pytest==7.1.2
yapf==0.32.0
pre-commit==2.20.0

[fastmri]
h5py==3.7.0
scikit_image==0.19.3

[full]
scikit-learn==1.0.1
h5py==3.7.0
scikit_image==0.19.3
jraph==0.0.6.dev0
scikit-learn==1.0.1
sentencepiece==0.1.97
tensorflow-text==2.9.0
sentencepiece==0.1.97
tensorflow-text==2.9.0
sacrebleu==1.3.1
pydub==0.25.1

[full_dev]
scikit-learn==1.0.1
h5py==3.7.0
scikit_image==0.19.3
jraph==0.0.6.dev0
scikit-learn==1.0.1
sentencepiece==0.1.97
tensorflow-text==2.9.0
sentencepiece==0.1.97
tensorflow-text==2.9.0
sacrebleu==1.3.1
pydub==0.25.1
isort==5.10.1
pylint==2.16.1
pytest==7.1.2
yapf==0.32.0
pre-commit==2.20.0

[jax_core_deps]
flax==0.6.10
optax==0.1.5
chex==0.1.7

[jax_cpu]
flax==0.6.10
optax==0.1.5
chex==0.1.7
jax==0.4.10
jaxlib==0.4.10

[jax_gpu]
flax==0.6.10
optax==0.1.5
chex==0.1.7
jax[cuda]==0.4.10
jaxlib==0.4.10+cuda11.cudnn86

[librispeech_conformer]
sentencepiece==0.1.97
tensorflow-text==2.9.0

[ogbg]
jraph==0.0.6.dev0
scikit-learn==1.0.1

[pytorch_cpu]
torch==1.13.0
torchvision==0.14.0

[pytorch_gpu]
torch==1.13.0+cu117
torchvision==0.14.0+cu117

[wandb]
wandb==0.13.4

[wmt]
sentencepiece==0.1.97
tensorflow-text==2.9.0
sacrebleu==1.3.1
