Docker for getting jax to work with cuda, for reproducing ml experiments like eicl. Sure, let's NOT make a compatibility matrix and let people fight for their lives on cuda
-
Updated
Aug 11, 2024 - Shell
#
jaxline
Here is 1 public repository matching this topic...
Improve this page
Add a description, image, and links to the jaxline topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the jaxline topic, visit your repo's landing page and select "manage topics."