Add SIGMOSFilterStage for audio quality assessment - #1577

Merged

sarahyurick merged 31 commits into

NVIDIA-NeMo:mainfrom

shubhamNvidia:pr/audio-sigmos

Apr 6, 2026

Commits on Mar 5, 2026

Add SIGMOSFilterStage and ensure_cudnn_loaded utility
shubhamNvidia
committed
2c80669
Add SIGMOS model files
shubhamNvidia
committed
5faad7a
Add unit tests for SIGMOSFilterStage
shubhamNvidia
committed
e64f75b

Commits on Mar 11, 2026

Simplified config, removed temp file creation, uses in-memory predict_audio_mos, simplified pipeline
shubhamNvidia
committed
f3f8928

Commits on Mar 12, 2026

Guard GPU fallback, delegate GPU detection to stage, replace print with logger
shubhamNvidia
committed
01dc37e

Commits on Mar 20, 2026

Fix copyrights, restore upstream gpu_utils and added ensure cudnn function, remove shebang and __main__, rewrite SIGMOS tests to match stage API
shubhamNvidia
committed
ffd5888

Commits on Mar 22, 2026

Commits on Mar 25, 2026

Commits on Mar 26, 2026

Commits on Mar 30, 2026

Add nested segments handling to SIGMOSFilterStage
shubhamNvidia
committed
250c67f

Commits on Mar 31, 2026

Merge branch 'main' into pr/audio-sigmos
sarahyurick
authored
6912606

Commits on Apr 2, 2026

Commits on Apr 3, 2026

Merge upstream/main, resolve conflicts keeping both changes
shubhamNvidia
committed
32215b4

Commits on Apr 5, 2026

Commits on Apr 6, 2026