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
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
Add a description, image, and links to the ml-engineering topic page so that developers can more easily learn about it.
To associate your repository with the ml-engineering topic, visit your repo's landing page and select "manage topics."