torch>=1.9.0
torchaudio>=0.9.0
transformers>=4.20.0
PyYAML>=5.4.0
tqdm>=4.60.0
numpy>=1.19.0
numba>=0.51.0
librosa
dac
descript-audiotools
snac
opencv-python

[all]
webdataset>=0.2.0
h5py>=3.0.0
tensorflow>=2.8.0
jax>=0.4.0
jaxlib>=0.4.0
grain>=0.1.0

[dev]
pytest>=6.0.0
pytest-cov>=2.10.0
black>=21.0.0
flake8>=3.8.0
mypy>=0.800
twine>=3.0.0
wheel>=0.36.0
build>=0.7.0

[grain]
grain>=0.1.0
jax>=0.4.0
jaxlib>=0.4.0

[hdf5]
h5py>=3.0.0

[jax]
jax>=0.4.0
jaxlib>=0.4.0

[tfrecord]
tensorflow>=2.8.0

[webdataset]
webdataset>=0.2.0
