aistore
fsspec>=2024.12.0
huggingface_hub>=0.24
numpy>=1.22
onnx>=1.7.0
scikit-learn
setuptools>=70.0.0
smart-open
tensorboard
text-unidecode
torch>=2.6.0
tqdm>=4.41.0
wrapt

[:platform_system != "Darwin"]
cuda-bindings

[:platform_system == "Darwin"]
numba

[all]
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
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
datasets>=3.2.0
einops
pandas
sentencepiece<1.0.0
braceexpand
kaldialign
lhotse>=1.33.0
librosa>=0.10.1
packaging
sacrebleu
scipy>=0.14
soundfile
whisper_normalizer
janome
jieba
librosa
matplotlib
nltk
pypinyin
pypinyin-dict
pyopenjtalk
librosa>=0.10.0
pystoi
nemo_automodel
flashoptim
peft<=0.18.0

[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

[asr]
braceexpand
einops
kaldialign
lhotse>=1.33.0
librosa>=0.10.1
packaging
sacrebleu
scipy>=0.14
soundfile
whisper_normalizer
datasets>=3.2.0
pandas
sentencepiece<1.0.0
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
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
einops
kaldialign
lhotse>=1.33.0
librosa>=0.10.1
packaging
sacrebleu
scipy>=0.14
soundfile
whisper_normalizer

[audio]
einops
lhotse>=1.33.0
librosa>=0.10.0
matplotlib
pystoi
scipy>=0.14
soundfile
datasets>=3.2.0
pandas
sentencepiece<1.0.0
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
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>=3.2.0
einops
pandas
sentencepiece<1.0.0
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
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>=3.2.0
einops
pandas
sentencepiece<1.0.0

[compiled]
onnx-ir==0.2.1
onnxscript==0.7.0
deep_ep==1.2.1
nv-grouped-gemm==1.1.4.post8
causal-conv1d==1.6.2.post1
mamba-ssm==2.3.2.post1
flash-attn==2.8.3
transformer-engine[core_cu13,pytorch]==2.15

[compiled-a100]
onnx-ir==0.2.1
onnxscript==0.7.0
nv-grouped-gemm==1.1.4.post8
causal-conv1d==1.6.2.post1
mamba-ssm==2.3.2.post1
flash-attn==2.8.3
transformer-engine[core_cu12,pytorch]==2.15

[core]
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
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

[cu12]

[cu12:platform_system != "Darwin"]
numba-cuda[cu12]
cuda-python<13,>=12

[cu12:sys_platform == "linux"]
torch==2.12.0+cu126

[cu13]

[cu13:platform_system != "Darwin"]
numba-cuda[cu13]
cuda-python<14,>=13

[cu13:sys_platform == "linux"]
torch==2.12.0+cu132

[lightning]
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
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

[speechlm2]
nemo_automodel
flashoptim
peft<=0.18.0
braceexpand
einops
kaldialign
lhotse>=1.33.0
librosa>=0.10.1
packaging
sacrebleu
scipy>=0.14
soundfile
whisper_normalizer
datasets>=3.2.0
pandas
sentencepiece<1.0.0
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
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
janome
jieba
librosa
matplotlib
nltk
pypinyin
pypinyin-dict
pyopenjtalk

[speechlm2-only]
nemo_automodel
flashoptim
peft<=0.18.0

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

[tts]
einops
janome
jieba
librosa
matplotlib
nltk
pandas
pypinyin
pypinyin-dict
pyopenjtalk
braceexpand
kaldialign
lhotse>=1.33.0
librosa>=0.10.1
packaging
sacrebleu
scipy>=0.14
soundfile
whisper_normalizer
datasets>=3.2.0
sentencepiece<1.0.0
hydra-core<=1.3.2,>1.3
lightning<=2.4.0,>2.2.1
omegaconf<=2.3
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
