Skip to content

Tags: wayneck/jax

Tags

jaxlib-v0.3.5

Toggle jaxlib-v0.3.5's commit message
Jaxlib release v0.3.5

jax-v0.3.5

Toggle jax-v0.3.5's commit message
Jax release v0.3.5

jax-v0.3.4

Toggle jax-v0.3.4's commit message
jax version 0.3.4

jax-v0.3.3

Toggle jax-v0.3.3's commit message
jax version 0.3.3

jax-v0.3.2

Toggle jax-v0.3.2's commit message
jax release v0.3.2

jaxlib-v0.3.2

Toggle jaxlib-v0.3.2's commit message
jaxlib release v0.3.2

jax-v0.3.1

Toggle jax-v0.3.1's commit message
Revert back to adding aval on Device buffers inside local_shards and …

…convert the cached property to just the normal property.

This slows down the pjit path because now you are paying the cost to create avals during runtime.

PiperOrigin-RevId: 429647845

jaxlib-v0.3.0

Toggle jaxlib-v0.3.0's commit message
Release jax and jaxlib 0.3.0 as per the new release process.

PiperOrigin-RevId: 427809845

jax-v0.3.0

Toggle jax-v0.3.0's commit message
Release jax and jaxlib 0.3.0 as per the new release process.

PiperOrigin-RevId: 427809845

test-jaxlib

Toggle test-jaxlib's commit message
Merge pull request jax-ml#4671 from romanngg:conv_local

PiperOrigin-RevId: 426282505