Repository navigation
Jwave is not compatible with cuda 12.8 #241
Description
First of all, I wanted to say that this repository is really great, and it has allowed me to simulate many things. The current issue is that I've changed my GPU, which is only compatible with CUDA 12.8 and above. Because of this, the necessary version of JAX doesn't work with the current version of jwave. I was wondering if it would be possible for you to update the repository so that jwave can continue to be used with more modern graphics cards. Best regards.
Activity
astanziola commentedon Mar 22, 2026
Thanks for the kind words and for reporting this!
This has been fixed.
jwavenow works with modern JAX versions (including CUDA 12.8+). The dependency chain was updated in #245, andjaxdfnow requiresjax>=0.9.0which has full CUDA 12.8 support.Closing as resolved. Let me know if you run into any other issues!