You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
You can install Jax with CUDA via pip; here are the instructions for version 0.4.2, which BayesNewton currently uses (see e.g. the Jax v0.4.2 README, as it differs a bit from the current version):
hi
How can I speed up training on GPU such as VariationalGP?
The text was updated successfully, but these errors were encountered: