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

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.