Tags: wayneck/jax
Tags
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
Release jax and jaxlib 0.3.0 as per the new release process. PiperOrigin-RevId: 427809845
Release jax and jaxlib 0.3.0 as per the new release process. PiperOrigin-RevId: 427809845
Merge pull request jax-ml#4671 from romanngg:conv_local PiperOrigin-RevId: 426282505