Skip to content

Tags: IvyZX/jax

Tags

jax-v0.3.13

jax version 0.3.13

jax-v0.3.12

Merge pull request jax-ml#10718 from hawkinsp:jaxfix

PiperOrigin-RevId: 448849099

jax-v0.3.11

Merge pull request jax-ml#10688 from sharadmv:color-by-default

PiperOrigin-RevId: 448723842

jaxlib-v0.3.10

Update TF commit for release

PiperOrigin-RevId: 446555288

jax-v0.3.10

Update TF commit for release

PiperOrigin-RevId: 446555288

jax-v0.3.9

Enable a number of tests on GPU.

In particular, pjit/xmap work on CPU these days.

PiperOrigin-RevId: 446085110

jax-v0.3.8

jax version 0.3.8

jaxlib-v0.3.7

Jaxlib release v0.3.7

jax-v0.3.7

Jax release v0.3.7

jax-v0.3.6

Upgrade the bazel version to 5.1.1

PiperOrigin-RevId: 441338363