Description
Project URL
https://pypi.org/project/jaxlib
Does this project already exist?
- Yes
New Limit
200MiB
Update issue title
- I have updated the title.
Which indexes
PyPI
About the project
jaxlib is the binary library for jax (https://github.com/jax-ml/jax), which is a major machine learning framework. It has been active since around 2019 on pypi.
Our pypi releases are already split into multiple plugin packages to minimize file sizes. For example, jaxlib does not contain any CUDA code, which instead lives in jax-cuda12-pjrt and jax-cuda12-plugin which are plugins for the main jax package. jaxlib contains only the binary code that is common to all users of JAX.
JAX and jaxlib are considerably smaller in binary size than similar frameworks (e.g., torch, tensorflow).
Reasons for the request
Individual wheels were slightly below 100MiB for some time, but the current release is now just over 100MiB (100.3MiB) leading to a broken release where only some of our wheels are uploaded. We can work to reduce our binary size but we have been uncomfortably close to the 100MiB limit for a long time.
If it were possible to address this promptly it would help us and our users.
Code of Conduct
- I agree to follow the PSF Code of Conduct