Use CUDA as extensions?

[scheidan]

Very exciting package!

CUDA is quite a heavy dependency, that cannot be used on any machine. Could we use a package extension instead?

[rsenne]

Hi @scheidan, thank you, glad you like it!

We definitely can make that happen and its on my roadmap. Out of curiosity, do you have plans to use the package? I'd call it very experimental at the moment and so any feedback you have would be great. As a note, the biggest speedups you will get will be on GPU for sure i.e., the space of models where one sees reliable speedups on CPU is real, but I'd call it very limited (why the code has CUDA as a hard dep rn). So, eventually I'd like ot make this package compatible with non nvidia GPUs

[rsenne]

also, i'm very actively working on this package, and as of now am focused mostly on the modeling interface / AD enhancements. So, I can't promise this will happen soon unless you wanted to make a PR for it :)