absl-py
array_record
click
dm-tree
etils[enp,epath]>=0.9.0
numpy
promise
protobuf>=3.20
psutil
requests>=2.19.0
tensorflow-metadata
termcolor
toml
tqdm
wrapt

[:python_version < "3.9"]
importlib_resources

[aflw2k3d]
scipy

[beir]
apache-beam

[ble_wind_field]
gcsfs
zarr

[c4]
apache-beam
gcld3
langdetect
nltk
tldextract

[c4_wsrs]
apache-beam

[cats_vs_dogs]
matplotlib

[colorectal_histology]
Pillow

[common_voice]
pydub

[dev]
dill
jax[cpu]
jupyter
pytest
pytest-shard
pytest-xdist
pandas
pydub
apache-beam
conllu
pyyaml
tensorflow-io
pylint>=2.6.0
yapf

[dev:python_version < "3.10"]
datasets

[duke_ultrasound]
scipy

[eurosat]
scikit-image
tifffile
imagecodecs

[groove]
pretty_midi
pydub

[gtzan]
pydub

[huggingface]
dill
jax[cpu]
jupyter
pytest
pytest-shard
pytest-xdist
pandas
pydub
conllu
pyyaml
tensorflow-io
scipy
gcsfs
zarr
gcld3
langdetect
nltk
tldextract
matplotlib
Pillow
scikit-image
tifffile
imagecodecs
pretty_midi
opencv-python
networkx
h5py
mwparserfromhell
mwxml
bs4
lxml
pycocotools
datasets

[huggingface:python_version < "3.10"]
envlogger

[imagenet2012_corrupted]
opencv-python
scikit-image
scipy

[librispeech]
pydub

[locomotion]

[locomotion:python_version < "3.10"]
envlogger

[lsun]
tensorflow-io

[matplotlib]
matplotlib

[nsynth]
crepe>=0.0.11
librosa
scikit-learn==0.20.3

[ogbg_molpcba]
pandas
networkx

[pet_finder]
pandas

[robonet]
h5py

[robosuite_panda_pick_place_can]

[robosuite_panda_pick_place_can:python_version < "3.10"]
envlogger

[smartwatch_gestures]
pandas

[svhn]
scipy

[tensorflow]
tensorflow>=2.1

[tensorflow-data-validation]
tensorflow-data-validation

[tests-all]
dill
jax[cpu]
jupyter
pytest
pytest-shard
pytest-xdist
pandas
pydub
apache-beam
conllu
pyyaml
tensorflow-io
scipy
gcsfs
zarr
gcld3
langdetect
nltk
tldextract
matplotlib
Pillow
scikit-image
tifffile
imagecodecs
pretty_midi
opencv-python
networkx
h5py
mwparserfromhell
mwxml
bs4
lxml
pycocotools

[tests-all:python_version < "3.10"]
datasets
envlogger

[the300w_lp]
scipy

[wider_face]
Pillow

[wiki_dialog]
apache-beam

[wikipedia]
apache-beam
mwparserfromhell
mwxml

[wsc273]
bs4
lxml

[youtube_vis]
pycocotools
