Skip to content

Updating JAX content

Nestor Demeure requested to merge nestordemeure/nersc.gitlab.io:jax into main

This PR fixes and updates all mentions of JAX in the documentation.

In particular, it updates the installation instruction (both the CUDA 11 and CUDA 12 instructions were broken). Note that the CUDA 11 instructions have been removed, I could not get a JAX release running without warning given our CUDNN_cuda11 version.

Submitters and reviewers please make sure

  • internal links are relative and to the .md file
  • avoid links named "here"
  • alt-text for images is descriptive
  • doesn't duplicate existing content
  • spell check
  • renders correctly
  • does not include any identifiable information (real usernames, project names, etc)

Merge request reports