Use CUDA as extensions? #59

Activity

1. rsenne commented on Jul 24, 2026

   

   rsenne

   on Jul 24, 2026

   Owner

   More actions

   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
2. rsenne commented on Jul 24, 2026

   

   rsenne

   on Jul 24, 2026

   Owner

   More actions

   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 :)
3. rsenne

   added

   enhancementNew feature or requestNew feature or request

   on Jul 24, 2026
4. rsenne

   added a commit that references this issue on Aug 24, 2026

   Make CUDA a package extension (#59)

   ...

   54ec42b
5. rsenne commented on Aug 27, 2026

   

   rsenne

   on Aug 27, 2026

   Owner

   More actions

   Hi @scheidan see #68! But am curious, do you have a desire to use this on non-CUDA GPUs or is it more so you can try it out on CPU and the CUDA dep is big cost/annoying (i agree)?
6. rsenne

   added 2 commits that reference this issue on Sep 13, 2026

   Make CUDA a package extension (#59)

   ...

   27928e5

   Make CUDA a package extension (#68)

   ...

   32de28f
7. rsenne

   closed this as completedin #68on Sep 14, 2026