21.2 Beta 2: Stem extraction GPU acceleration

[FL-User]

Would it be too difficult to implement GPU acceleration, at least for Nvidia GPUs to the stem extraction? I've been using the demucs cli for a while and when I found out I could use it with my GPU I reduced the time it took from several minutes to a couple of seconds. I understand that PyTorch is only available with CUDA when it comes to hardware acceleration :/

[flipflopp99902]

surprised this doesnt have many other comments....

[therealkevinkevinson]

Definitely need this, and kind of confusing tha...