← all repositories

lucidrains/BS-RoFormer

PyTorch implementation of Band Split Roformer, a state-of-the-art attention network for separating music into individual stems.

BS-RoFormer
Velocity · 7d
+0.8
★ / day
Trend
steady
star history

This repository provides an implementation of the Band Split Roformer architecture developed by ByteDance AI Labs for music source separation. The model uses axial attention across frequency bands and time dimensions, along with rotary positional encoding, to achieve state-of-the-art performance in separating audio into stems such as vocals, drums, bass, and other instruments. It supports stereo training and multi-stem output.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.