Music Source Separation Python Packages

bs-roformer

Implementation of Band Split Roformer, SOTA Attention network for music source separation out of ByteDance AI Labs

torch-log-wmse

logWMSE, an audio quality metric & loss function with support for digital silence target. Useful for training and evaluating audio source separation systems.

hunterformsbs

Flexible frequency-band splitter for music source separation. The BandSplitRoformer module supports BSRoformer, MelBandRoformer, and custom overlapping or non-overlapping band configurations. Fully typed, modular, and documented, including migration help, usage, and paper references. PyTorch; CUDA-accelerated.

torch-l1-snr

Variations of L1 SNR Loss function for training audio source separation machine learning models

moises-light

Unofficial PyTorch implementation of "Moises-Light: Resource-efficient Band-split U-Net For Music Source Separation"

torch-log-wmse-audio-quality

logWMSE, an audio quality metric & loss function with support for digital silence target. Useful for training and evaluating audio source separation systems.

torchsubband

Pytorch implementation of subband decomposition

mlx-audio-separator

MLX-native audio stem separation for Apple Silicon

rope-conformer

A clean, plug-and-play Conformer encoder with rotary positional embeddings.

unmixer

Create and explore isolated tracks from music files

musion

MUSION, a toolbox for music information retrieval and music signal processing.