orbax
orbax copied to clipboard
Resubmit single array error message
Resubmit of https://github.com/google/orbax/pull/1304 with style adjustments.
Oh actually maybe this was merged in https://github.com/google/orbax/pull/1324? In which case I guess close this.