Actions: google/brax
Actions
110 workflow runs
110 workflow runs
jax.device_put_sharded and `jax.devic…
CI
#559:
Commit 303d89d
pushed
by
copybara-service
bot
jax.config.pmap_shmap_merge.
CI
#537:
Commit 7120ee3
pushed
by
copybara-service
bot