cerabinowitz

CUDA and Jax version

1

Hi, Thank you so much for this repo! What cuda version were you using when you trained with GPUs and did you use a specific Jax version as well? Thank...