Mac GPU
Hello,
First of all, thank you very much for your package. It has been beneficial to make it easier to create an SSM in my research. However, we use Macs in our lab, which means we cannot use the torch kernel's GPU capabilities. I have used torch.gpu before in much simpler Python codes on a Mac. So could you tell me where I should alter your code to make this possible? In particular, the instances where you define the torch.device(). Combing through it with a fine comb has proven to be a frustrating endeavour. Any help would be appreciated, and I will, of course, share the code with you if I get it working.
Thank you very much, Best wishes, Simão