Why separate installation? JAX with CUDA support is Linux-specific and requires system CUDA 12.1-12.9 pre-installed. Separating the installation avoids dependency ...

You can create a release to package software, along with release notes and links to binary files, for other people to use. Learn more about releases in our docs.