fsspec==2024.12.0
huggingface_hub>=0.24
numba==0.61.0
numpy<2.0.0,>=1.22
onnx>=1.7.0
protobuf==4.24.4
python-dateutil
ruamel.yaml
scikit-learn
setuptools>=70.0.0
tensorboard
text-unidecode
torch
tqdm>=4.41.0
wget
wrapt

[all]
black~=24.3
click>=8.1
coverage
isort<6.0.0,>5.1.0
parameterized
pytest
pytest-mock
pytest-runner
ruamel.yaml
sphinx
sphinxcontrib-bibtex
wandb
wget
wrapt
cloudpickle
fiddle
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
peft
torchmetrics>=0.11.0
transformers<=4.52.0,>=4.51.0
webdataset>=0.2.86
datasets
einops
inflect
mediapy==1.1.6
pandas
sacremoses>=0.0.43
sentencepiece<1.0.0
braceexpand
editdistance
g2p_en
jiwer<4.0.0,>=3.1.0
kaldi-python-io
kaldiio
lhotse>=1.26.0
librosa>=0.10.1
marshmallow
optuna
packaging
pyannote.core
pyannote.metrics
pydub
pyloudnorm
resampy
scipy>=0.14
soundfile
sox<=1.5.0
texterrors<1.0.0
num2words
numpy<2.0.0
accelerated-scan
boto3
faiss-cpu
flask_restful
ftfy
gdown
h5py
ijson
jieba
markdown2
matplotlib>=3.3.2
megatron_core
nltk>=3.6.5
numpy<2
opencc
pangu
prettytable
rapidfuzz
rouge_score
sacrebleu
sentence_transformers
tiktoken==0.7.0
unstructured==0.14.9
zarr<3.0.0,>=2.18.2
attrdict
cdifflib==1.2.6
janome
kornia
librosa
matplotlib
nltk
pypinyin
pypinyin-dict
seaborn
jiwer<4.0.0,>=2.0.0
progress>=1.5
tabulate>=0.8.7
textdistance>=4.1.5
tqdm
addict
clip
diffusers>=0.19.3
einops_exts
imageio
megatron-energon==5.2.0
nerfacc>=0.5.3
open_clip_torch==2.24.0
qwen_vl_utils
taming-transformers
torchdiffeq
torchsde
trimesh
lhotse>=1.22.0
librosa>=0.10.0
pystoi
accelerate
fastapi
pydantic-settings
uvicorn
nvidia-lm-eval==25.5

[all:"arm" not in platform_machine and "aarch" not in platform_machine and sys_platform != "darwin"]
nemo_text_processing

[all:platform_machine != "x86_64" or platform_system != "Darwin"]
pesq

[all:platform_machine == "x86_64" and platform_system != "Darwin"]
bitsandbytes==0.45.3

[all:platform_system != "Darwin"]
nvidia-modelopt[torch]<=0.27.1,>=0.23.2
nvidia-resiliency-ext<1.0.0,>=0.3.0
tensorstore<0.1.72
nvidia-pytriton

[all:sys_platform == "linux" and platform_machine == "x86_64"]
decord

[asr]
braceexpand
editdistance
einops
g2p_en
jiwer<4.0.0,>=3.1.0
kaldi-python-io
kaldiio
lhotse>=1.26.0
librosa>=0.10.1
marshmallow
optuna
packaging
pyannote.core
pyannote.metrics
pydub
pyloudnorm
resampy
ruamel.yaml
scipy>=0.14
soundfile
sox<=1.5.0
texterrors<1.0.0
cloudpickle
fiddle
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
peft
torchmetrics>=0.11.0
transformers<=4.52.0,>=4.51.0
wandb
webdataset>=0.2.86
datasets
inflect
mediapy==1.1.6
pandas
sacremoses>=0.0.43
sentencepiece<1.0.0
num2words
numpy<2.0.0

[asr:platform_machine == "x86_64" and platform_system != "Darwin"]
bitsandbytes==0.45.3

[audio]
einops
lhotse>=1.22.0
librosa>=0.10.0
matplotlib
pystoi
scipy>=0.14
soundfile
cloudpickle
fiddle
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
peft
torchmetrics>=0.11.0
transformers<=4.52.0,>=4.51.0
wandb
webdataset>=0.2.86
datasets
inflect
mediapy==1.1.6
pandas
sacremoses>=0.0.43
sentencepiece<1.0.0

[audio:platform_machine != "x86_64" or platform_system != "Darwin"]
pesq

[audio:platform_machine == "x86_64" and platform_system != "Darwin"]
bitsandbytes==0.45.3

[common]
datasets
einops
inflect
mediapy==1.1.6
pandas
sacremoses>=0.0.43
sentencepiece<1.0.0
cloudpickle
fiddle
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
peft
torchmetrics>=0.11.0
transformers<=4.52.0,>=4.51.0
wandb
webdataset>=0.2.86

[common:platform_machine == "x86_64" and platform_system != "Darwin"]
bitsandbytes==0.45.3

[core]
cloudpickle
fiddle
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
peft
torchmetrics>=0.11.0
transformers<=4.52.0,>=4.51.0
wandb
webdataset>=0.2.86

[core:platform_machine == "x86_64" and platform_system != "Darwin"]
bitsandbytes==0.45.3

[ctc_segmentation]
num2words
numpy<2.0.0

[deploy]
accelerate
fastapi
pydantic-settings
uvicorn
zarr<3.0.0,>=2.18.2

[deploy:platform_system != "Darwin"]
nvidia-pytriton
tensorstore<0.1.72

[eval]
nvidia-lm-eval==25.5

[llm]
accelerated-scan
boto3
faiss-cpu
flask_restful
ftfy
gdown
h5py
ijson
jieba
markdown2
matplotlib>=3.3.2
megatron_core
nltk>=3.6.5
numpy<2
opencc
pangu
prettytable
rapidfuzz
rouge_score
sacrebleu
sentence_transformers
tiktoken==0.7.0
unstructured==0.14.9
zarr<3.0.0,>=2.18.2
cloudpickle
fiddle
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
peft
torchmetrics>=0.11.0
transformers<=4.52.0,>=4.51.0
wandb
webdataset>=0.2.86
datasets
einops
inflect
mediapy==1.1.6
pandas
sacremoses>=0.0.43
sentencepiece<1.0.0

[llm:platform_machine == "x86_64" and platform_system != "Darwin"]
bitsandbytes==0.45.3

[llm:platform_system != "Darwin"]
nvidia-modelopt[torch]<=0.27.1,>=0.23.2
nvidia-resiliency-ext<1.0.0,>=0.3.0
tensorstore<0.1.72

[multimodal]
addict
clip
diffusers>=0.19.3
einops_exts
imageio
kornia
megatron-energon==5.2.0
nerfacc>=0.5.3
open_clip_torch==2.24.0
qwen_vl_utils
taming-transformers
torchdiffeq
torchsde
trimesh
accelerated-scan
boto3
faiss-cpu
flask_restful
ftfy
gdown
h5py
ijson
jieba
markdown2
matplotlib>=3.3.2
megatron_core
nltk>=3.6.5
numpy<2
opencc
pangu
prettytable
rapidfuzz
rouge_score
sacrebleu
sentence_transformers
tiktoken==0.7.0
unstructured==0.14.9
zarr<3.0.0,>=2.18.2
cloudpickle
fiddle
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
peft
torchmetrics>=0.11.0
transformers<=4.52.0,>=4.51.0
wandb
webdataset>=0.2.86
datasets
einops
inflect
mediapy==1.1.6
pandas
sacremoses>=0.0.43
sentencepiece<1.0.0

[multimodal:platform_machine == "x86_64" and platform_system != "Darwin"]
bitsandbytes==0.45.3

[multimodal:platform_system != "Darwin"]
nvidia-modelopt[torch]<=0.27.1,>=0.23.2
nvidia-resiliency-ext<1.0.0,>=0.3.0
tensorstore<0.1.72

[multimodal:sys_platform == "linux" and platform_machine == "x86_64"]
decord

[nlp]
accelerated-scan
boto3
faiss-cpu
flask_restful
ftfy
gdown
h5py
ijson
jieba
markdown2
matplotlib>=3.3.2
megatron_core
nltk>=3.6.5
numpy<2
opencc
pangu
prettytable
rapidfuzz
rouge_score
sacrebleu
sentence_transformers
tiktoken==0.7.0
unstructured==0.14.9
zarr<3.0.0,>=2.18.2
cloudpickle
fiddle
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
peft
torchmetrics>=0.11.0
transformers<=4.52.0,>=4.51.0
wandb
webdataset>=0.2.86
datasets
einops
inflect
mediapy==1.1.6
pandas
sacremoses>=0.0.43
sentencepiece<1.0.0

[nlp:platform_machine == "x86_64" and platform_system != "Darwin"]
bitsandbytes==0.45.3

[nlp:platform_system != "Darwin"]
nvidia-modelopt[torch]<=0.27.1,>=0.23.2
nvidia-resiliency-ext<1.0.0,>=0.3.0
tensorstore<0.1.72

[slu]
jiwer<4.0.0,>=2.0.0
progress>=1.5
tabulate>=0.8.7
textdistance>=4.1.5
tqdm
braceexpand
editdistance
einops
g2p_en
jiwer<4.0.0,>=3.1.0
kaldi-python-io
kaldiio
lhotse>=1.26.0
librosa>=0.10.1
marshmallow
optuna
packaging
pyannote.core
pyannote.metrics
pydub
pyloudnorm
resampy
ruamel.yaml
scipy>=0.14
soundfile
sox<=1.5.0
texterrors<1.0.0
cloudpickle
fiddle
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
peft
torchmetrics>=0.11.0
transformers<=4.52.0,>=4.51.0
wandb
webdataset>=0.2.86
datasets
inflect
mediapy==1.1.6
pandas
sacremoses>=0.0.43
sentencepiece<1.0.0
num2words
numpy<2.0.0

[slu:platform_machine == "x86_64" and platform_system != "Darwin"]
bitsandbytes==0.45.3

[test]
attrdict
cdifflib==1.2.6
einops
janome
jieba
kornia
librosa
matplotlib
nltk
pandas
pypinyin
pypinyin-dict
seaborn
cloudpickle
fiddle
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
peft
torchmetrics>=0.11.0
transformers<=4.52.0,>=4.51.0
wandb
webdataset>=0.2.86
datasets
inflect
mediapy==1.1.6
sacremoses>=0.0.43
sentencepiece<1.0.0

[test:"arm" not in platform_machine and "aarch" not in platform_machine and sys_platform != "darwin"]
nemo_text_processing

[test:platform_machine == "x86_64" and platform_system != "Darwin"]
bitsandbytes==0.45.3

[tts]
attrdict
cdifflib==1.2.6
einops
janome
jieba
kornia
librosa
matplotlib
nltk
pandas
pypinyin
pypinyin-dict
seaborn
cloudpickle
fiddle
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
peft
torchmetrics>=0.11.0
transformers<=4.52.0,>=4.51.0
wandb
webdataset>=0.2.86
datasets
inflect
mediapy==1.1.6
sacremoses>=0.0.43
sentencepiece<1.0.0
braceexpand
editdistance
g2p_en
jiwer<4.0.0,>=3.1.0
kaldi-python-io
kaldiio
lhotse>=1.26.0
librosa>=0.10.1
marshmallow
optuna
packaging
pyannote.core
pyannote.metrics
pydub
pyloudnorm
resampy
ruamel.yaml
scipy>=0.14
soundfile
sox<=1.5.0
texterrors<1.0.0
num2words
numpy<2.0.0

[tts:"arm" not in platform_machine and "aarch" not in platform_machine and sys_platform != "darwin"]
nemo_text_processing

[tts:platform_machine == "x86_64" and platform_system != "Darwin"]
bitsandbytes==0.45.3
