Colin Gaffney
Results
2
issues of
Colin Gaffney
Internal change.
Resubmit of https://github.com/google/orbax/pull/1304 with style adjustments.