Inconsistency in requirements (PyPI vs conda)

[camposandro]

A version incompatibility issue came up while I was trying to release a new package (EzTaoX) on conda-forge that depends on optax (conda-forge/staged-recipes#32095):

optax 0.2.6 has requirement jax>=0.5.3, but you have jax 0.4.31.
optax 0.2.6 has requirement jaxlib>=0.5.3, but you have jaxlib 0.4.31.

This appears to be caused by differing JAX dependency pins between the conda-forge recipe and the PyPI release. The latest version v0.2.6 requires jax>=0.5.3. However, the conda recipe still pins jax according to v0.2.5, jax>=0.4.27.

[rdyro]

What's your recommendation here?

[camposandro]

I think we can sync the versions of jax/jaxlib on the conda-forge recipe to match those on the pyproject.toml, and generate a new conda build (number 1) for v0.2.6. If I am not mistaken, conda will use that new build as the preferred one. I can also open a PR with these changes if you think they are reasonable!

[rdyro]

I'm not sure how the conda forge recipe is created. What does this mean on our side / what should we do?

[camposandro]

I opened PR #28 on the conda feedstock. I think this should do it!

[Sumu004]

I've opened a PR on the conda-forge feedstock to fix this: conda-forge/optax-feedstock#31

The recipe for v0.2.8 still pinned jax>=0.4.27 / jaxlib>=0.4.27 (from the 0.2.5 era). The PR updates them to jax>=0.5.3 / jaxlib>=0.5.3 to match the current pyproject.toml, and bumps the build number to 1 so conda picks up the corrected metadata.