python3Packages.jax[lib]: 0.6.2 -> 0.7.1 #427588

GaetanLepage · 2025-07-22T22:49:20Z

Things done

Diff: jax-ml/jax@jax-v0.6.2...jax-v0.7.1
Changelogs:

Add a 👍 reaction to pull requests you find important.

samuela

diff lgtm

GaetanLepage · 2025-08-08T19:24:44Z

`nixpkgs-review` result

Generated using nixpkgs-review.

Command: nixpkgs-review pr 427588
Commit: e03950bc0713f05a84564cb86068c60fbd46af2d

`x86_64-linux`

❌ 32 packages failed to build:

python312Packages.arviz
python312Packages.arviz.dist
python312Packages.bambi
python312Packages.bambi.dist
python312Packages.blackjax
python312Packages.blackjax.dist
python312Packages.corner
python312Packages.corner.dist
python312Packages.distrax
python312Packages.distrax.dist
python312Packages.equinox
python312Packages.equinox.dist
python312Packages.flowjax
python312Packages.flowjax.dist
python312Packages.flowmc
python312Packages.flowmc.dist
python312Packages.lineax
python312Packages.lineax.dist
python312Packages.numpyro
python312Packages.numpyro.dist
python312Packages.nutpie
python312Packages.nutpie.dist
python312Packages.optimistix
python312Packages.optimistix.dist
python312Packages.paramax
python312Packages.paramax.dist
python312Packages.pymc
python312Packages.pymc.dist
python312Packages.rlax
python312Packages.rlax.dist
python313Packages.blackjax
python313Packages.blackjax.dist

✅ 144 packages built:

exo
exo.dist
python312Packages.ale-py
python312Packages.ale-py.dist
python312Packages.anndata
python312Packages.anndata.dist
python312Packages.array-api-compat
python312Packages.array-api-compat.dist
python312Packages.augmax
python312Packages.augmax.dist
python312Packages.brax
python312Packages.brax.dist
python312Packages.chex
python312Packages.chex.dist
python312Packages.cleanlab
python312Packages.cleanlab.dist
python312Packages.dalle-mini
python312Packages.dalle-mini.dist
python312Packages.deepface
python312Packages.deepface.dist
python312Packages.dm-haiku
python312Packages.dm-haiku.dist
python312Packages.dm-sonnet
python312Packages.dm-sonnet.dist
python312Packages.edward
python312Packages.edward.dist
python312Packages.evosax
python312Packages.evosax.dist
python312Packages.flax
python312Packages.flax.dist
python312Packages.gymnasium
python312Packages.gymnasium.dist
python312Packages.jax
python312Packages.jax-cuda12-pjrt
python312Packages.jax-cuda12-plugin
python312Packages.jax.dist
python312Packages.jaxlib (python312Packages.jaxlib-bin)
python312Packages.jaxlib.dist (python312Packages.jaxlib-bin.dist)
python312Packages.jaxopt
python312Packages.jaxopt.dist
python312Packages.jmp
python312Packages.jmp.dist
python312Packages.keras
python312Packages.keras.dist
python312Packages.mhcflurry
python312Packages.mhcflurry.dist
python312Packages.minari
python312Packages.minari.dist
python312Packages.mtcnn
python312Packages.mtcnn.dist
python312Packages.mujoco-mjx
python312Packages.mujoco-mjx.dist
python312Packages.nifty8
python312Packages.nifty8.dist
python312Packages.optax
python312Packages.optax.dist
python312Packages.optax.testsout
python312Packages.orbax-checkpoint
python312Packages.orbax-checkpoint.dist
python312Packages.pettingzoo
python312Packages.pettingzoo.dist
python312Packages.pymanopt
python312Packages.pymanopt.dist
python312Packages.pytensor
python312Packages.pytensor.dist
python312Packages.pyvers
python312Packages.pyvers.dist
python312Packages.retinaface
python312Packages.retinaface.dist
python312Packages.shimmy
python312Packages.shimmy.dist
python312Packages.skrl
python312Packages.skrl.dist
python312Packages.stable-baselines3
python312Packages.stable-baselines3.dist
python312Packages.tensordict
python312Packages.tensordict.dist
python312Packages.tensorflow-datasets
python312Packages.tensorflow-datasets.dist
python312Packages.tensorflow-probability
python312Packages.tensorflow-probability.dist
python312Packages.tianshou
python312Packages.tianshou.dist
python312Packages.tinygrad
python312Packages.tinygrad.dist
python312Packages.treescope
python312Packages.treescope.dist
python312Packages.tyro
python312Packages.tyro.dist
python312Packages.unsloth
python312Packages.unsloth-zoo
python312Packages.unsloth-zoo.dist
python312Packages.unsloth.dist
python312Packages.viser
python312Packages.viser.dist
python312Packages.vqgan-jax
python312Packages.vqgan-jax.dist
python312Packages.waymax
python312Packages.waymax.dist
python313Packages.anndata
python313Packages.anndata.dist
python313Packages.array-api-compat
python313Packages.array-api-compat.dist
python313Packages.augmax
python313Packages.augmax.dist
python313Packages.chex
python313Packages.chex.dist
python313Packages.equinox
python313Packages.equinox.dist
python313Packages.flowmc
python313Packages.flowmc.dist
python313Packages.jax
python313Packages.jax-cuda12-pjrt
python313Packages.jax-cuda12-plugin
python313Packages.jax.dist
python313Packages.jaxlib (python313Packages.jaxlib-bin)
python313Packages.jaxlib.dist (python313Packages.jaxlib-bin.dist)
python313Packages.jaxopt
python313Packages.jaxopt.dist
python313Packages.jmp
python313Packages.jmp.dist
python313Packages.lineax
python313Packages.lineax.dist
python313Packages.mujoco-mjx
python313Packages.mujoco-mjx.dist
python313Packages.nifty8
python313Packages.nifty8.dist
python313Packages.optax
python313Packages.optax.dist
python313Packages.optax.testsout
python313Packages.optimistix
python313Packages.optimistix.dist
python313Packages.orbax-checkpoint
python313Packages.orbax-checkpoint.dist
python313Packages.paramax
python313Packages.paramax.dist
python313Packages.pyvers
python313Packages.pyvers.dist
python313Packages.tensordict
python313Packages.tensordict.dist
python313Packages.tinygrad
python313Packages.tinygrad.dist
python313Packages.treescope
python313Packages.treescope.dist

Error logs: `x86_64-linux`

python312Packages.blackjax

  772  773  774  775  776  777  778  779  780  781  782  783  784  785
  786  787  788  789  790  791  792  793  794  795  796  797  798  799
  800  801  802  803  804  805  806  807  808  809  810  811  812  813
  814  815  816  817  818  819  820  821  822  823  824  825  826  827
  828  829  830  831  832  833  834  835  836  837  838  839  840  841
  842  843  844  845  846  847  848  849  850  851  852  853  854  855
  856  857  858  859  860  861  862  863  864  865  866  867  868  869
  870  871  872  873  874  875  876  877  878  879  880  881  882  883
  884  885  886  887  888  889  890  891  892  893  894  895  896  897
  898  899  900  901  902  903  904  905  906  907  908  909  910  911
  912  913  914  915  916  917  918  919  920  921  922  923  924  925
  926  927  928  929  930  931  932  933  934  935  936  937  938  939
  940  941  942  943  944  945  946  947  948  949  950  951  952  953
  954  955  956  957  958  959  960  961  962  963  964  965  966  967
  968  969  970  971  972  973  974  975  976  977  978  979  980  981
  982  983  984  985  986  987  988  989  990  991  992  993  994  995
  996  997  998  999 1000 1001]
=========================== short test summary info ============================
FAILED tests/adaptation/test_adaptation.py::test_chees_adaptation[adaptation_filters0] - AssertionError: 
============= 1 failed, 461 passed, 1 skipped in 68.17s (0:01:08) ==============

python312Packages.distrax

FAILED distrax/_src/distributions/quantized_test.py::QuantizedSurvivalFunctionConsistencyTest::test_log_survival_function_log_cdf_consistency_noop cutoffs__with_device - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/distributions/von_mises_test.py::VonMisesTest::test_with_two_distributions_cross-ent distrax_to_distrax__without_device - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/utils/importance_sampling_test.py::ImportanceSamplingTest::test_importance_sampling_ratios_off_policy__without_device - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/utils/importance_sampling_test.py::ImportanceSamplingTest::test_importance_sampling_ratios_off_policy__with_jit - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/distributions/von_mises_test.py::VonMisesTest::test_with_two_distributions_kl distrax_to_distrax__with_jit - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/utils/jittable_test.py::JittableTest::test_traceable - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/distributions/von_mises_test.py::VonMisesTest::test_with_two_distributions_cross-ent distrax_to_distrax__with_device - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/utils/importance_sampling_test.py::ImportanceSamplingTest::test_importance_sampling_ratios_on_policy__with_device - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/utils/jittable_test.py::JittableTest::test_different_jittables_to_compiled_function - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/utils/importance_sampling_test.py::ImportanceSamplingTest::test_importance_sampling_ratios_on_policy__without_device - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/utils/jittable_test.py::JittableTest::test_jittable - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/utils/jittable_test.py::JittableTest::test_modifying_object_data_does_not_leak_tracers - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/utils/importance_sampling_test.py::ImportanceSamplingTest::test_importance_sampling_ratios_on_policy__with_jit - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/distributions/von_mises_test.py::VonMisesTest::test_with_two_distributions_cross-ent distrax_to_distrax__with_jit - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/utils/jittable_test.py::JittableTest::test_metadata_modification_statements_are_removed_by_compilation - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/distributions/von_mises_test.py::VonMisesTest::test_with_two_distributions_kl distrax_to_distrax__with_device - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/distributions/von_mises_test.py::VonMisesTest::test_with_two_distributions_kl distrax_to_distrax__without_device - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/utils/jittable_test.py::JittableTest::test_vmappable - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
FAILED distrax/_src/utils/importance_sampling_test.py::ImportanceSamplingTest::test_importance_sampling_ratios_off_policy__with_device - AttributeError: jax.interpreters.xla.abstractify was deprecated in JAX v0.5...
== 402 failed, 5313 passed, 1025 skipped, 2323 warnings in 183.23s (0:03:03) ===

python312Packages.equinox

            # Linearize so that we save residuals
            return jax.linearize(loop, init_xs)
    size = 100_000
    # nontrivial batch size is important to ensure that the `.at[].set()` is really a
    # scatter, and that XLA doesn't optimise it into a dynamic_update_slice. (Which
    # can be switched with `select` in the compiler.)
    args = jnp.array([0, 1]), jnp.zeros((2, size))
    f(*args)  # compile

    speed = timeit.timeit(lambda: f(*args), number=1)


  assert speed < 4


E       assert 4.725688154809177 < 4
tests/test_while_loop.py:514: AssertionError

----------------------------- Captured stdout call -----------------------------

stop_gradient_perturb_val (False, False, False, (False, True))

=========================== short test summary info ============================

FAILED tests/test_while_loop.py::test_speed_buffer_while[True] - assert 4.725688154809177 < 4

============= 1 failed, 452 passed, 1 skipped in 70.17s (0:01:10) ==============

python313Packages.blackjax

  772  773  774  775  776  777  778  779  780  781  782  783  784  785
  786  787  788  789  790  791  792  793  794  795  796  797  798  799
  800  801  802  803  804  805  806  807  808  809  810  811  812  813
  814  815  816  817  818  819  820  821  822  823  824  825  826  827
  828  829  830  831  832  833  834  835  836  837  838  839  840  841
  842  843  844  845  846  847  848  849  850  851  852  853  854  855
  856  857  858  859  860  861  862  863  864  865  866  867  868  869
  870  871  872  873  874  875  876  877  878  879  880  881  882  883
  884  885  886  887  888  889  890  891  892  893  894  895  896  897
  898  899  900  901  902  903  904  905  906  907  908  909  910  911
  912  913  914  915  916  917  918  919  920  921  922  923  924  925
  926  927  928  929  930  931  932  933  934  935  936  937  938  939
  940  941  942  943  944  945  946  947  948  949  950  951  952  953
  954  955  956  957  958  959  960  961  962  963  964  965  966  967
  968  969  970  971  972  973  974  975  976  977  978  979  980  981
  982  983  984  985  986  987  988  989  990  991  992  993  994  995
  996  997  998  999 1000 1001]
=========================== short test summary info ============================
FAILED tests/adaptation/test_adaptation.py::test_chees_adaptation[adaptation_filters0] - AssertionError: 
============= 1 failed, 461 passed, 1 skipped in 188.52s (0:03:08) =============

GaetanLepage · 2025-08-18T13:10:16Z

`nixpkgs-review` result

Generated using nixpkgs-review.

Command: nixpkgs-review pr 427588
Commit: 8d3370fd592c28c25e4d783ca7d88dc5351ed996

`x86_64-linux`

✅ 176 packages built:

exo
exo.dist
python312Packages.ale-py
python312Packages.ale-py.dist
python312Packages.anndata
python312Packages.anndata.dist
python312Packages.array-api-compat
python312Packages.array-api-compat.dist
python312Packages.arviz
python312Packages.arviz.dist
python312Packages.augmax
python312Packages.augmax.dist
python312Packages.bambi
python312Packages.bambi.dist
python312Packages.blackjax
python312Packages.blackjax.dist
python312Packages.brax
python312Packages.brax.dist
python312Packages.chex
python312Packages.chex.dist
python312Packages.cleanlab
python312Packages.cleanlab.dist
python312Packages.corner
python312Packages.corner.dist
python312Packages.dalle-mini
python312Packages.dalle-mini.dist
python312Packages.deepface
python312Packages.deepface.dist
python312Packages.distrax
python312Packages.distrax.dist
python312Packages.dm-haiku
python312Packages.dm-haiku.dist
python312Packages.dm-sonnet
python312Packages.dm-sonnet.dist
python312Packages.edward
python312Packages.edward.dist
python312Packages.equinox
python312Packages.equinox.dist
python312Packages.evosax
python312Packages.evosax.dist
python312Packages.flax
python312Packages.flax.dist
python312Packages.flowjax
python312Packages.flowjax.dist
python312Packages.flowmc
python312Packages.flowmc.dist
python312Packages.gymnasium
python312Packages.gymnasium.dist
python312Packages.jax
python312Packages.jax-cuda12-pjrt
python312Packages.jax-cuda12-plugin
python312Packages.jax.dist
python312Packages.jaxlib (python312Packages.jaxlib-bin)
python312Packages.jaxlib.dist (python312Packages.jaxlib-bin.dist)
python312Packages.jaxopt
python312Packages.jaxopt.dist
python312Packages.jmp
python312Packages.jmp.dist
python312Packages.keras
python312Packages.keras.dist
python312Packages.lineax
python312Packages.lineax.dist
python312Packages.mhcflurry
python312Packages.mhcflurry.dist
python312Packages.minari
python312Packages.minari.dist
python312Packages.mtcnn
python312Packages.mtcnn.dist
python312Packages.mujoco-mjx
python312Packages.mujoco-mjx.dist
python312Packages.nifty8
python312Packages.nifty8.dist
python312Packages.numpyro
python312Packages.numpyro.dist
python312Packages.nutpie
python312Packages.nutpie.dist
python312Packages.optax
python312Packages.optax.dist
python312Packages.optax.testsout
python312Packages.optimistix
python312Packages.optimistix.dist
python312Packages.orbax-checkpoint
python312Packages.orbax-checkpoint.dist
python312Packages.paramax
python312Packages.paramax.dist
python312Packages.pettingzoo
python312Packages.pettingzoo.dist
python312Packages.pymanopt
python312Packages.pymanopt.dist
python312Packages.pymc
python312Packages.pymc.dist
python312Packages.pytensor
python312Packages.pytensor.dist
python312Packages.pyvers
python312Packages.pyvers.dist
python312Packages.retinaface
python312Packages.retinaface.dist
python312Packages.rlax
python312Packages.rlax.dist
python312Packages.shimmy
python312Packages.shimmy.dist
python312Packages.skrl
python312Packages.skrl.dist
python312Packages.stable-baselines3
python312Packages.stable-baselines3.dist
python312Packages.tensordict
python312Packages.tensordict.dist
python312Packages.tensorflow-datasets
python312Packages.tensorflow-datasets.dist
python312Packages.tensorflow-probability
python312Packages.tensorflow-probability.dist
python312Packages.tianshou
python312Packages.tianshou.dist
python312Packages.tinygrad
python312Packages.tinygrad.dist
python312Packages.treescope
python312Packages.treescope.dist
python312Packages.tyro
python312Packages.tyro.dist
python312Packages.unsloth
python312Packages.unsloth-zoo
python312Packages.unsloth-zoo.dist
python312Packages.unsloth.dist
python312Packages.viser
python312Packages.viser.dist
python312Packages.vqgan-jax
python312Packages.vqgan-jax.dist
python312Packages.waymax
python312Packages.waymax.dist
python313Packages.anndata
python313Packages.anndata.dist
python313Packages.array-api-compat
python313Packages.array-api-compat.dist
python313Packages.augmax
python313Packages.augmax.dist
python313Packages.blackjax
python313Packages.blackjax.dist
python313Packages.chex
python313Packages.chex.dist
python313Packages.equinox
python313Packages.equinox.dist
python313Packages.flowmc
python313Packages.flowmc.dist
python313Packages.jax
python313Packages.jax-cuda12-pjrt
python313Packages.jax-cuda12-plugin
python313Packages.jax.dist
python313Packages.jaxlib (python313Packages.jaxlib-bin)
python313Packages.jaxlib.dist (python313Packages.jaxlib-bin.dist)
python313Packages.jaxopt
python313Packages.jaxopt.dist
python313Packages.jmp
python313Packages.jmp.dist
python313Packages.lineax
python313Packages.lineax.dist
python313Packages.mujoco-mjx
python313Packages.mujoco-mjx.dist
python313Packages.nifty8
python313Packages.nifty8.dist
python313Packages.optax
python313Packages.optax.dist
python313Packages.optax.testsout
python313Packages.optimistix
python313Packages.optimistix.dist
python313Packages.orbax-checkpoint
python313Packages.orbax-checkpoint.dist
python313Packages.paramax
python313Packages.paramax.dist
python313Packages.pyvers
python313Packages.pyvers.dist
python313Packages.tensordict
python313Packages.tensordict.dist
python313Packages.tinygrad
python313Packages.tinygrad.dist
python313Packages.treescope
python313Packages.treescope.dist

GaetanLepage · 2025-08-18T13:55:58Z

`nixpkgs-review` result

Generated using nixpkgs-review.

Command: nixpkgs-review pr 427588 --package python3Packages.jax
Commit: 8d3370fd592c28c25e4d783ca7d88dc5351ed996

`x86_64-linux`

✅ 2 packages built:

python3Packages.jax
python3Packages.jax.dist (python3Packages.jax.dist.dist)

`aarch64-linux`

✅ 2 packages built:

python3Packages.jax
python3Packages.jax.dist (python3Packages.jax.dist.dist)

`x86_64-darwin`

⏩ 2 packages marked as broken and skipped:

python3Packages.jax
python3Packages.jax.dist

`aarch64-darwin`

❌ 2 packages failed to build:

python3Packages.jax
python3Packages.jax.dist (python3Packages.jax.dist.dist)

Error logs: `aarch64-darwin`

python3Packages.jax

       [ 0.32755727 , -0.16292901 ,  0.36086023 ],
       [-0.89855    , -0.9306742  ,  0.30420253 ],
       [ 0.4700789  ,  0.72444236 , -0.040413253]], dtype=float32)
FAILED tests/linalg_test.py::ScipyLinalgTest::testSolveTriangular9 - AssertionError: 
Not equal to tolerance rtol=0.0001, atol=1e-06
Mismatched elements: 160 / 160 (100%)

Max absolute difference among violations: 9.270169

Max relative difference among violations: 476.554

ACTUAL: array([[[0.5993011  , 0.1481528  , 4.0326166  , 0.59425926 ,

0.15692338 , 0.30325112 , 3.796802   , 0.1714465  ,

1.2916036  , 1.0908637  ],...

DESIRED: array([[[-0.08125915 , -0.25842887 , -0.008479829, -0.15118487 ,

-0.28587827 , -0.47274005 , -0.23783009 , -0.37468398 ,

0.00729001 , -0.16181852 ],...

FAILED tests/linalg_test.py::LaxLinalgTest::testEigh3 - AssertionError: np.float32(14.160136) not less than or equal to np.float32(0.013507687)

FAILED tests/linalg_test.py::LaxLinalgTest::testEigh5 - AssertionError: np.float32(7.058923) not less than or equal to np.float32(0.0075361016)

FAILED tests/linalg_test.py::LaxLinalgTest::testEigh7 - AssertionError: np.float32(9.275328) not less than or equal to np.float32(0.009189508)

FAILED tests/svd_test.py::SvdTest::testSvdWithOnRankDeficientInputZeroColumns1 - RuntimeWarning: overflow encountered in dot

========= 75 failed, 29265 passed, 19362 skipped in 1418.60s (0:23:38) =========

samuela · 2025-08-21T00:28:16Z

nit: typo in 8778c7f commit description

GaetanLepage · 2025-08-21T07:08:24Z

nit: typo in 8778c7f commit description

Thanks! Fixed. I still have to understand why so many numerical assertions are completely wrong now on darwin...

samuela · 2025-08-21T18:49:12Z

Yeah, I also find that peculiar. Not your fault at least

GaetanLepage · 2025-08-29T09:40:48Z

Yeah, I also find that peculiar. Not your fault at least

Reported upstream: jax-ml/jax#31428

And I skipped the relevant test files on darwin.

pkgs/development/python-modules/jax-cuda12-pjrt/default.nix

NickCao

Diff LGTM, please run nixpkgs-review.

GaetanLepage · 2025-08-30T00:05:31Z

`nixpkgs-review` result

Generated using nixpkgs-review.

Command: nixpkgs-review pr 427588 --package python312Packages.jax --package python3Packages.jax
Commit: ce3dde9506e91b675dcc1bc05c1b0819436cb6af

`x86_64-linux`

✅ 4 packages built:

python312Packages.jax
python312Packages.jax.dist (python312Packages.jax.dist.dist)
python3Packages.jax
python3Packages.jax.dist (python3Packages.jax.dist.dist)

`aarch64-linux`

✅ 4 packages built:

python312Packages.jax
python312Packages.jax.dist (python312Packages.jax.dist.dist)
python3Packages.jax
python3Packages.jax.dist (python3Packages.jax.dist.dist)

`x86_64-darwin`

⏩ 4 packages marked as broken and skipped:

python312Packages.jax
python312Packages.jax.dist
python3Packages.jax
python3Packages.jax.dist

`aarch64-darwin`

✅ 4 packages built:

python312Packages.jax
python312Packages.jax.dist (python312Packages.jax.dist.dist)
python3Packages.jax
python3Packages.jax.dist (python3Packages.jax.dist.dist)

Diff: jax-ml/jax@jax-v0.6.2...jax-v0.7.0 Changelog: https://github.com/jax-ml/jax/releases/tag/jax-v0.7.0

…ibility

blackjax-devs/blackjax#795

GaetanLepage · 2025-08-30T14:27:03Z

`nixpkgs-review` result

Generated using nixpkgs-review.

Command: nixpkgs-review pr 427588
Commit: 37aa265ca7a7e2e55acc946b755858cdc08900c1

`x86_64-linux`

❌ 2 packages failed to build:

python312Packages.cleanlab
python312Packages.cleanlab.dist

✅ 162 packages built:

exo
exo.dist
python312Packages.ale-py
python312Packages.ale-py.dist
python312Packages.anndata
python312Packages.anndata.dist
python312Packages.array-api-compat
python312Packages.array-api-compat.dist
python312Packages.augmax
python312Packages.augmax.dist
python312Packages.blackjax
python312Packages.blackjax.dist
python312Packages.brax
python312Packages.brax.dist
python312Packages.chex
python312Packages.chex.dist
python312Packages.dalle-mini
python312Packages.dalle-mini.dist
python312Packages.deepface
python312Packages.deepface.dist
python312Packages.dm-haiku
python312Packages.dm-haiku.dist
python312Packages.dm-sonnet
python312Packages.dm-sonnet.dist
python312Packages.edward
python312Packages.edward.dist
python312Packages.equinox
python312Packages.equinox.dist
python312Packages.evosax
python312Packages.evosax.dist
python312Packages.flax
python312Packages.flax.dist
python312Packages.flowmc
python312Packages.flowmc.dist
python312Packages.gymnasium
python312Packages.gymnasium.dist
python312Packages.jax
python312Packages.jax-cuda12-pjrt
python312Packages.jax-cuda12-plugin
python312Packages.jax.dist
python312Packages.jaxlib (python312Packages.jaxlib-bin)
python312Packages.jaxlib.dist (python312Packages.jaxlib-bin.dist)
python312Packages.jaxopt
python312Packages.jaxopt.dist
python312Packages.jmp
python312Packages.jmp.dist
python312Packages.keras
python312Packages.keras.dist
python312Packages.kserve
python312Packages.kserve.dist
python312Packages.lineax
python312Packages.lineax.dist
python312Packages.mhcflurry
python312Packages.mhcflurry.dist
python312Packages.minari
python312Packages.minari.dist
python312Packages.mtcnn
python312Packages.mtcnn.dist
python312Packages.mujoco-mjx
python312Packages.mujoco-mjx.dist
python312Packages.nifty8
python312Packages.nifty8.dist
python312Packages.optax
python312Packages.optax.dist
python312Packages.optax.testsout
python312Packages.optimistix
python312Packages.optimistix.dist
python312Packages.orbax-checkpoint
python312Packages.orbax-checkpoint.dist
python312Packages.outlines
python312Packages.outlines.dist
python312Packages.paramax
python312Packages.paramax.dist
python312Packages.pettingzoo
python312Packages.pettingzoo.dist
python312Packages.pymanopt
python312Packages.pymanopt.dist
python312Packages.pyvers
python312Packages.pyvers.dist
python312Packages.retinaface
python312Packages.retinaface.dist
python312Packages.scanpy
python312Packages.scanpy.dist
python312Packages.shimmy
python312Packages.shimmy.dist
python312Packages.skrl
python312Packages.skrl.dist
python312Packages.stable-baselines3
python312Packages.stable-baselines3.dist
python312Packages.tensordict
python312Packages.tensordict.dist
python312Packages.tensorflow-datasets
python312Packages.tensorflow-datasets.dist
python312Packages.tianshou
python312Packages.tianshou.dist
python312Packages.tinygrad
python312Packages.tinygrad.dist
python312Packages.torchrl
python312Packages.torchrl.dist
python312Packages.treescope
python312Packages.treescope.dist
python312Packages.tyro
python312Packages.tyro.dist
python312Packages.unsloth
python312Packages.unsloth-zoo
python312Packages.unsloth-zoo.dist
python312Packages.unsloth.dist
vllm (python312Packages.vllm)
vllm.dist (python312Packages.vllm.dist)
python312Packages.vqgan-jax
python312Packages.vqgan-jax.dist
python312Packages.waymax
python312Packages.waymax.dist
python313Packages.anndata
python313Packages.anndata.dist
python313Packages.array-api-compat
python313Packages.array-api-compat.dist
python313Packages.augmax
python313Packages.augmax.dist
python313Packages.blackjax
python313Packages.blackjax.dist
python313Packages.chex
python313Packages.chex.dist
python313Packages.equinox
python313Packages.equinox.dist
python313Packages.flowmc
python313Packages.flowmc.dist
python313Packages.jax
python313Packages.jax-cuda12-pjrt
python313Packages.jax-cuda12-plugin
python313Packages.jax.dist
python313Packages.jaxlib (python313Packages.jaxlib-bin)
python313Packages.jaxlib.dist (python313Packages.jaxlib-bin.dist)
python313Packages.jaxopt
python313Packages.jaxopt.dist
python313Packages.jmp
python313Packages.jmp.dist
python313Packages.lineax
python313Packages.lineax.dist
python313Packages.mujoco-mjx
python313Packages.mujoco-mjx.dist
python313Packages.nifty8
python313Packages.nifty8.dist
python313Packages.optax
python313Packages.optax.dist
python313Packages.optax.testsout
python313Packages.optimistix
python313Packages.optimistix.dist
python313Packages.orbax-checkpoint
python313Packages.orbax-checkpoint.dist
python313Packages.paramax
python313Packages.paramax.dist
python313Packages.pyvers
python313Packages.pyvers.dist
python313Packages.scanpy
python313Packages.scanpy.dist
python313Packages.tensordict
python313Packages.tensordict.dist
python313Packages.tinygrad
python313Packages.tinygrad.dist
python313Packages.treescope
python313Packages.treescope.dist

Error logs: `x86_64-linux`

python312Packages.cleanlab

ERROR tests/datalab/test_issue_finder.py::TestIssueFinder::test_underperforming_group_availability_issue_1065[keys=['pred_probs', 'features', 'cluster_ids'] -> available] - AssertionError
ERROR tests/datalab/test_issue_finder.py::TestIssueFinder::test_underperforming_group_availability_issue_1065[keys=['pred_probs', 'knn_graph', 'cluster_ids'] -> available] - AssertionError
ERROR tests/datalab/test_issue_finder.py::TestIssueFinder::test_underperforming_group_availability_issue_1065[keys=['pred_probs', 'features', 'knn_graph', 'cluster_ids'] -> available] - AssertionError
ERROR tests/datalab/test_issue_finder.py::TestIssueFinder::test_underperforming_group_availability_issue_1065[keys=['pred_probs'] -> unavailable] - AssertionError
ERROR tests/datalab/test_issue_finder.py::TestIssueFinder::test_get_available_issue_types - AssertionError
ERROR tests/datalab/test_issue_finder.py::TestIssueFinder::test_find_issues - AssertionError
ERROR tests/datalab/test_issue_finder.py::TestIssueFinder::test_validate_issue_types_dict - AssertionError
ERROR tests/datalab/test_issue_finder.py::TestRegressionIssueFinder::test_get_available_issue_types - AssertionError
ERROR tests/datalab/test_report.py::TestReporter::test_init - AssertionError
ERROR tests/datalab/test_report.py::TestReporter::test_report - AssertionError
ERROR tests/datalab/test_report.py::TestReporter::test_get_report[True] - AssertionError
ERROR tests/datalab/test_report.py::TestReporter::test_get_report[False] - AssertionError
ERROR tests/datalab/test_report.py::TestReporter::test_show_all_issues[True-Here is a lab summary\n\nfoo report\n\n\nbar report] - AssertionError
ERROR tests/datalab/test_report.py::TestReporter::test_show_all_issues[False-Here is a lab summary\n\nfoo report] - AssertionError
ERROR tests/datalab/test_report.py::TestReporter::test_summary_with_score - AssertionError
ERROR tests/datalab/test_report.py::TestReporter::test_summary_without_score - AssertionError
ERROR tests/spurious_correlation/test_spurious_correlation.py::TestImagelabReporterAdapter::test_report[dark] - AssertionError
ERROR tests/spurious_correlation/test_spurious_correlation.py::TestImagelabReporterAdapter::test_report[odd_aspect_ratio] - AssertionError
ERROR tests/spurious_correlation/test_spurious_correlation.py::TestImagelabReporterAdapter::test_report[identity] - AssertionError
= 151 failed, 652 passed, 15 deselected, 2 xfailed, 26968 warnings, 233 errors in 578.79s (0:09:38) =

GaetanLepage · 2025-08-30T14:41:51Z

cleanlab also fails on master (unrelated).

samuela

Thanks for putting this together @GaetanLepage ! Any remaining blockers to merge?

pkgs/development/python-modules/distrax/default.nix

…atibility

Diff: jax-ml/jax@jax-v0.7.0...jax-v0.7.1 Changelog: https://github.com/jax-ml/jax/releases/tag/jax-v0.7.1

patrick-kidger/equinox#1086 patrick-kidger/equinox#1082

keras-team/keras#21629

patrick-kidger/optimistix#160

GaetanLepage · 2025-08-30T19:53:01Z

Thanks for putting this together @GaetanLepage ! Any remaining blockers to merge?

Nope, if you're fine with the diff, feel free to press the button. This was a big one!

GaetanLepage force-pushed the update/python3Packages.jax branch from d4b533b to 5558bde Compare July 22, 2025 22:51

nixpkgs-ci bot added 10.rebuild-linux: 11-100 This PR causes between 11 and 100 packages to rebuild on Linux. 10.rebuild-darwin: 11-100 This PR causes between 11 and 100 packages to rebuild on Darwin. 6.topic: python Python is a high-level, general-purpose programming language. labels Jul 22, 2025

nix-owners bot requested review from samuela and natsukium July 22, 2025 22:58

nixpkgs-ci bot added 9.needs: reviewer This PR currently has no reviewers requested and needs attention. and removed 9.needs: reviewer This PR currently has no reviewers requested and needs attention. labels Jul 22, 2025

nix-owners bot requested a review from ndl July 23, 2025 13:45

GaetanLepage force-pushed the update/python3Packages.jax branch from 1656027 to 9761641 Compare July 24, 2025 07:58

samuela approved these changes Jul 28, 2025

View reviewed changes

nixpkgs-ci bot added 12.approvals: 1 This PR was reviewed and approved by one person. 12.approved-by: package-maintainer This PR was reviewed and approved by a maintainer listed in any of the changed packages. labels Jul 28, 2025

GaetanLepage force-pushed the update/python3Packages.jax branch 2 times, most recently from b1c06d5 to e03950b Compare August 8, 2025 10:48

GaetanLepage mentioned this pull request Aug 18, 2025

test_chees_adaptation fails with latest jax 0.7.0 blackjax-devs/blackjax#795

Open

GaetanLepage force-pushed the update/python3Packages.jax branch from e03950b to c6e8558 Compare August 18, 2025 08:37

nix-owners bot requested review from bcdarwin and onny August 18, 2025 08:43

GaetanLepage requested a review from samuela August 18, 2025 13:10

GaetanLepage force-pushed the update/python3Packages.jax branch 2 times, most recently from 71c6e47 to d9ba35d Compare August 20, 2025 21:02

GaetanLepage changed the title ~~python3Packages.jax[lib]: 0.6.2 -> 0.7.0~~ python3Packages.jax[lib]: 0.6.2 -> 0.7.1 Aug 20, 2025

GaetanLepage force-pushed the update/python3Packages.jax branch from d9ba35d to 4b03fb7 Compare August 21, 2025 07:07

GaetanLepage force-pushed the update/python3Packages.jax branch from 4a7a7cb to 820eee4 Compare August 29, 2025 09:32

GaetanLepage requested a review from NickCao August 29, 2025 21:38

NickCao reviewed Aug 29, 2025

View reviewed changes

pkgs/development/python-modules/jax-cuda12-pjrt/default.nix Outdated Show resolved Hide resolved

GaetanLepage force-pushed the update/python3Packages.jax branch from 820eee4 to ce3dde9 Compare August 29, 2025 21:58

NickCao reviewed Aug 29, 2025

View reviewed changes

GaetanLepage force-pushed the update/python3Packages.jax branch 2 times, most recently from 38d615f to fd62135 Compare August 30, 2025 09:16

GaetanLepage added 4 commits August 30, 2025 12:50

python3Packages.jax[lib]: 0.6.2 -> 0.7.0

2f680b8

Diff: jax-ml/jax@jax-v0.6.2...jax-v0.7.0 Changelog: https://github.com/jax-ml/jax/releases/tag/jax-v0.7.0

python3Packages.tensorflow-probability: patch to fix jax 0.7.0 compat…

9a0a4c0

…ibility

python3Packages.chex: remove unused input pythonOlder

317bd97

python3Packages.blackjax: skip failing test

c2bead3

blackjax-devs/blackjax#795

This comment was marked as outdated.

Sign in to view

GaetanLepage force-pushed the update/python3Packages.jax branch from fd62135 to 4746763 Compare August 30, 2025 12:47

GaetanLepage requested a review from NickCao August 30, 2025 14:41

samuela approved these changes Aug 30, 2025

View reviewed changes

pkgs/development/python-modules/distrax/default.nix Outdated Show resolved Hide resolved

GaetanLepage added 5 commits August 30, 2025 21:52

python3Packages.distrax: fetch upstream patches to fix jax 0.7.0 comp…

352643a

…atibility

python3Packages.jax[lib]: 0.7.0 -> 0.7.1

4d76031

Diff: jax-ml/jax@jax-v0.7.0...jax-v0.7.1 Changelog: https://github.com/jax-ml/jax/releases/tag/jax-v0.7.1

python3Packages.equinox: backport patches to fix with jax 0.7.0

7d5f6f6

patrick-kidger/equinox#1086 patrick-kidger/equinox#1082

python3Packages.keras: skip failing tests

ea8ae9f

keras-team/keras#21629

python3Packages.optimistix: disable failing test

4cfbd1d

patrick-kidger/optimistix#160

GaetanLepage force-pushed the update/python3Packages.jax branch from 37aa265 to 4cfbd1d Compare August 30, 2025 19:52

samuela merged commit e250e01 into NixOS:master Aug 30, 2025
28 of 31 checks passed

GaetanLepage deleted the update/python3Packages.jax branch August 30, 2025 22:45

Uh oh!

python3Packages.jax[lib]: 0.6.2 -> 0.7.1 #427588

python3Packages.jax[lib]: 0.6.2 -> 0.7.1 #427588

Uh oh!

Conversation

GaetanLepage commented Jul 22, 2025 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Things done

Uh oh!

samuela left a comment

Choose a reason for hiding this comment

Uh oh!

GaetanLepage commented Aug 8, 2025

nixpkgs-review result

x86_64-linux

Uh oh!

GaetanLepage commented Aug 18, 2025

nixpkgs-review result

x86_64-linux

Uh oh!

GaetanLepage commented Aug 18, 2025

nixpkgs-review result

x86_64-linux

aarch64-linux

x86_64-darwin

aarch64-darwin

Uh oh!

samuela commented Aug 21, 2025

Uh oh!

GaetanLepage commented Aug 21, 2025

Uh oh!

samuela commented Aug 21, 2025

Uh oh!

GaetanLepage commented Aug 29, 2025

Uh oh!

Uh oh!

NickCao left a comment

Choose a reason for hiding this comment

Uh oh!

GaetanLepage commented Aug 30, 2025

nixpkgs-review result

x86_64-linux

aarch64-linux

x86_64-darwin

aarch64-darwin

Uh oh!

This comment was marked as outdated.

GaetanLepage commented Aug 30, 2025

nixpkgs-review result

x86_64-linux

Uh oh!

GaetanLepage commented Aug 30, 2025

Uh oh!

samuela left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

GaetanLepage commented Aug 30, 2025

Uh oh!

Uh oh!

Uh oh!

GaetanLepage commented Jul 22, 2025 •

edited

Loading

`nixpkgs-review` result

`x86_64-linux`

`nixpkgs-review` result

`x86_64-linux`

`nixpkgs-review` result

`x86_64-linux`

`aarch64-linux`

`x86_64-darwin`

`aarch64-darwin`

`nixpkgs-review` result

`x86_64-linux`

`aarch64-linux`

`x86_64-darwin`

`aarch64-darwin`

`nixpkgs-review` result

`x86_64-linux`