Allow auto-registration to fall back to function outputs when no loss tags are registered. by copybara-service[bot] · Pull Request #423 · google-deepmind/kfac-jax

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

@copybara-service

The following contributors were found for this pull request:

✅ 086619d PR Opener: @copybara-service[bot]
✅ 086619d Author: @james-martens <jam******ens@google.com>

(Only the first commit for a unique contributor is listed.)