Code

Making KFAC use pjit instead of pmap#121

Code

Open

copybara-service[bot] wants to merge 1 commit intomaingoogle-deepmind/kfac-jax:mainfrom test_526595657google-deepmind/kfac-jax:test_526595657Copy head branch name to clipboard

+626-534

Commits

Commits on May 5, 2023

• Making KFAC use pjit instead of pmap

  Show description for 1c5ed39
  

  botev

  authored and

  KfacJaxDev

  committed
  1c5ed39

  Copy full SHA for 1c5ed39