Bump `jax` and `jaxlib` versions #366

Thomas-Christie · 2023-08-25T15:23:40Z

Type of changes

[N/A] I've formatted the new code by running poetry run pre-commit run --all-files --show-diff-on-failure before committing.
[N/A] I've added tests for new code.
[N/A] I've added docstrings for the new code.

In response to issue (jax-ml/jax#15951) causing us to pin jaxlib being fixed. In order to perform this update I ran the following commands:

poetry remove jax
poetry remove jaxlib
poetry add "jax>=0.4.10"  
poetry add "jaxlib>=0.4.10"
poetry lock --no-update

Issue Number: #336

In response to issue (jax-ml/jax#15951) causing us to pin `jaxlib` being fixed.

Bump jax and jaxlib versions

4e18cc4

In response to issue (jax-ml/jax#15951) causing us to pin `jaxlib` being fixed.

Thomas-Christie requested a review from thomaspinder August 25, 2023 15:25

Bump orbax-checkpoint version

02207b3

thomaspinder approved these changes Aug 25, 2023

View reviewed changes

thomaspinder merged commit c5eb47b into JaxGaussianProcesses:main Aug 25, 2023

Thomas-Christie mentioned this pull request Aug 25, 2023

bug: Overly restrictive jaxlib dependency? #336

Closed

Thomas-Christie deleted the jaxlib-update branch August 26, 2023 11:20