iglee / jax-cuda-eicl-exp-docker Goto Github PK
View Code? Open in Web Editor NEWDocker 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