Non-CUDA support?

[ppp-one]

Congrats on the paper! You mentioned porting to other frameworks - which ones do you have in mind/timeline?

[leigh2]

Hi Peter,
Probably opencl (but open to suggestions). Hopefully there's not too much of a performance hit. I don't want to have to maintain too many different frameworks (ideally <=2). Bear in mind that I've not started, have no experience with opencl, and it's not my primary focus, so it may take some time. We can leave this issue open if you like and I'll update it in future.
Leigh

[ppp-one]

PyTorch with MPS Support would be my initial thoughts given most astronomers are mac users. But OpenCL would work too from my readings. However, I'm no expert here. Curious what comes down the line :)