We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
cuda11_cudnn82
cuda=11,cudnn=82
pip install jax[tpu]