File Limit Request: jax-cuda13-pjrt - 200 MB #6967
Description
Project URL
https://pypi.org/project/jax-cuda13-pjrt/
Does this project already exist?
- Yes
New Limit
200 MB
Update issue title
- I have updated the title.
Which indexes
PyPI
About the project
Similar to jax-cuda12-pjrt, this wheel provides NVIDIA GPU support as a plugin for https://pypi.org/project/jax/ for upcoming CUDA 13.
GPU support is split into a separate wheel precisely to reduce pypi space. jax and jaxlib are the packages that contain most of jax, whereas jax-cuda13-pjrt and jax-cuda13-plugin contain NVIDIA GPU support as a plugin.
Reasons for the request
Since the lateset jax-cuda12-pjrt wheel is about 126MB, we expect jax-cuda13-pjrt to also exceed 100MB.
This default file size limit will block jax-cuda13-pjrt releases. We need to raise cuda13's file size limit to 200MB, similar to cuda12.
Code of Conduct
- I agree to follow the PSF Code of Conduct