fsspec==2024.12.0
huggingface_hub>=0.24
numexpr<2.14.0
numpy>=1.22
onnx>=1.7.0
protobuf~=5.29.5
python-dateutil
ruamel.yaml
scikit-learn
setuptools>=70.0.0
tensorboard
text-unidecode
torch
tqdm>=4.41.0
wget
wrapt

[:platform_system != "Darwin"]
numba-cuda==0.15.1
cuda_bindings==13.0.0

[:platform_system == "Darwin"]
numba

[all]
black~=24.3
click>=8.1
coverage
isort<6.0.0,>5.1.0
parameterized
pytest
pytest-httpserver
pytest-mock
pytest-runner
ruamel.yaml
sphinx
sphinxcontrib-bibtex
wandb
wget
wrapt
nemo_run
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
webdataset>=0.2.86
nv_one_logger_core>=2.3.1
nv_one_logger_training_telemetry>=2.3.1
nv_one_logger_pytorch_lightning_integration>=2.3.1
datasets
einops
inflect
mediapy==1.1.6
pandas
sacremoses>=0.0.43
sentencepiece<1.0.0
braceexpand
diskcache
editdistance
jiwer<4.0.0,>=3.1.0
kaldi-python-io
kaldialign<=0.9.1
lhotse>=1.32.0
librosa>=0.10.1
marshmallow
optuna
packaging
pyannote.core
pyannote.metrics
pydub
pyloudnorm
resampy
scipy>=0.14
soundfile
sox<=1.5.0
whisper_normalizer
accelerated-scan
boto3
faiss-cpu
flask_restful
ftfy
gdown
h5py
ijson
jieba
markdown2
matplotlib>=3.3.2
megatron_core
multi-storage-client>=0.21.0
nltk>=3.6.5
numpy<2
nvidia-modelopt==0.37.0
nvtx
opencc
pangu
prettytable
rapidfuzz
rouge_score
sacrebleu
sentence_transformers
tiktoken==0.7.0
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.31.1
librosa>=0.10.0
pystoi

[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_system != "Darwin"]
nvidia-resiliency-ext<1.0.0,>=0.3.0
tensorstore<0.1.72

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

[asr]
braceexpand
diskcache
editdistance
einops
jiwer<4.0.0,>=3.1.0
kaldi-python-io
kaldialign<=0.9.1
lhotse>=1.32.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
whisper_normalizer
datasets
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
wandb
webdataset>=0.2.86
nv_one_logger_core>=2.3.1
nv_one_logger_training_telemetry>=2.3.1
nv_one_logger_pytorch_lightning_integration>=2.3.1

[asr-only]
braceexpand
diskcache
editdistance
einops
jiwer<4.0.0,>=3.1.0
kaldi-python-io
kaldialign<=0.9.1
lhotse>=1.32.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
whisper_normalizer

[audio]
einops
lhotse>=1.31.1
librosa>=0.10.0
matplotlib
pystoi
scipy>=0.14
soundfile
datasets
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
wandb
webdataset>=0.2.86
nv_one_logger_core>=2.3.1
nv_one_logger_training_telemetry>=2.3.1
nv_one_logger_pytorch_lightning_integration>=2.3.1

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

[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
wandb
webdataset>=0.2.86
nv_one_logger_core>=2.3.1
nv_one_logger_training_telemetry>=2.3.1
nv_one_logger_pytorch_lightning_integration>=2.3.1

[common-only]
datasets
einops
inflect
mediapy==1.1.6
pandas
sacremoses>=0.0.43
sentencepiece<1.0.0

[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
wandb
webdataset>=0.2.86
nv_one_logger_core>=2.3.1
nv_one_logger_training_telemetry>=2.3.1
nv_one_logger_pytorch_lightning_integration>=2.3.1

[lightning]
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
wandb
webdataset>=0.2.86
nv_one_logger_core>=2.3.1
nv_one_logger_training_telemetry>=2.3.1
nv_one_logger_pytorch_lightning_integration>=2.3.1

[llm]
accelerated-scan
boto3
faiss-cpu
flask_restful
ftfy
gdown
h5py
ijson
jieba
markdown2
matplotlib>=3.3.2
megatron_core
multi-storage-client>=0.21.0
nltk>=3.6.5
numpy<2
nvidia-modelopt==0.37.0
nvtx
opencc
pangu
prettytable
rapidfuzz
rouge_score
sacrebleu
sentence_transformers
tiktoken==0.7.0
zarr<3.0.0,>=2.18.2
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
wandb
webdataset>=0.2.86
nv_one_logger_core>=2.3.1
nv_one_logger_training_telemetry>=2.3.1
nv_one_logger_pytorch_lightning_integration>=2.3.1

[llm:platform_system != "Darwin"]
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
multi-storage-client>=0.21.0
nltk>=3.6.5
numpy<2
nvidia-modelopt==0.37.0
nvtx
opencc
pangu
prettytable
rapidfuzz
rouge_score
sacrebleu
sentence_transformers
tiktoken==0.7.0
zarr<3.0.0,>=2.18.2
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
wandb
webdataset>=0.2.86
nv_one_logger_core>=2.3.1
nv_one_logger_training_telemetry>=2.3.1
nv_one_logger_pytorch_lightning_integration>=2.3.1

[multimodal-only]
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

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

[multimodal:platform_system != "Darwin"]
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
multi-storage-client>=0.21.0
nltk>=3.6.5
numpy<2
nvidia-modelopt==0.37.0
nvtx
opencc
pangu
prettytable
rapidfuzz
rouge_score
sacrebleu
sentence_transformers
tiktoken==0.7.0
zarr<3.0.0,>=2.18.2
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
wandb
webdataset>=0.2.86
nv_one_logger_core>=2.3.1
nv_one_logger_training_telemetry>=2.3.1
nv_one_logger_pytorch_lightning_integration>=2.3.1

[nlp-only]
accelerated-scan
boto3
faiss-cpu
flask_restful
ftfy
gdown
h5py
ijson
jieba
markdown2
matplotlib>=3.3.2
megatron_core
multi-storage-client>=0.21.0
nltk>=3.6.5
numpy<2
nvidia-modelopt==0.37.0
nvtx
opencc
pangu
prettytable
rapidfuzz
rouge_score
sacrebleu
sentence_transformers
tiktoken==0.7.0
zarr<3.0.0,>=2.18.2

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

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

[run]
nemo_run

[slu]
jiwer<4.0.0,>=2.0.0
progress>=1.5
tabulate>=0.8.7
textdistance>=4.1.5
tqdm
braceexpand
diskcache
editdistance
einops
jiwer<4.0.0,>=3.1.0
kaldi-python-io
kaldialign<=0.9.1
lhotse>=1.32.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
whisper_normalizer
datasets
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
wandb
webdataset>=0.2.86
nv_one_logger_core>=2.3.1
nv_one_logger_training_telemetry>=2.3.1
nv_one_logger_pytorch_lightning_integration>=2.3.1

[test]
black~=24.3
click>=8.1
coverage
isort<6.0.0,>5.1.0
parameterized
pytest
pytest-httpserver
pytest-mock
pytest-runner
ruamel.yaml
sphinx
sphinxcontrib-bibtex
wandb
wget
wrapt
attrdict
cdifflib==1.2.6
einops
janome
jieba
kornia
librosa
matplotlib
nltk
pandas
pypinyin
pypinyin-dict
seaborn
datasets
inflect
mediapy==1.1.6
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
webdataset>=0.2.86
nv_one_logger_core>=2.3.1
nv_one_logger_training_telemetry>=2.3.1
nv_one_logger_pytorch_lightning_integration>=2.3.1

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

[tts]
attrdict
cdifflib==1.2.6
einops
janome
jieba
kornia
librosa
matplotlib
nltk
pandas
pypinyin
pypinyin-dict
seaborn
braceexpand
diskcache
editdistance
jiwer<4.0.0,>=3.1.0
kaldi-python-io
kaldialign<=0.9.1
lhotse>=1.32.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
whisper_normalizer
datasets
inflect
mediapy==1.1.6
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
wandb
webdataset>=0.2.86
nv_one_logger_core>=2.3.1
nv_one_logger_training_telemetry>=2.3.1
nv_one_logger_pytorch_lightning_integration>=2.3.1

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