mouung

Results 1 comments of mouung

Problem solved! I upgraded Cuda to 11.7 and re-installed Jax with Cuda support. Now the calculation is flying! Thanks for the help!