Actions: google-deepmind/kfac-jax
Actions
Showing runs from all workflows
923 workflow runs
923 workflow runs
jax_pmap_shmap_merge=False to the top of t…
ci
#1837:
Commit 406b21f
pushed
by
copybara-service
bot
jax_pmap_shmap_merge=False to the top of the optimizer.py file. The new pmap implementation currently seems to slow down projects using kfac_jax.
ci
#1836:
Pull request #367
synchronize
by
copybara-service
bot
jax_pmap_shmap_merge=False to the top of the optimizer.py file. The new pmap implementation currently seems to slow down projects using kfac_jax.
ci
#1835:
Pull request #367
synchronize
by
copybara-service
bot
jax_pmap_shmap_merge=False to the top of the optimizer.py file. The new pmap implementation currently seems to slow down projects using kfac_jax.
ci
#1834:
Pull request #367
synchronize
by
copybara-service
bot