Fix multi-device NaN check in KFAC training step - #104

Fix multi-device NaN check in KFAC training step#104

copybara-service[bot] merged 1 commit into

google-deepmind:maingoogle-deepmind/ferminet:mainfrom

Qianruipku:hotfixQianruipku/ferminet:hotfixCopy head branch name to clipboard