orbax
orbax copied to clipboard
google
Orbax provides common utility libraries for JAX users.
![copybara-service[bot] avatar](https://avatars.githubusercontent.com/in/44061?v=4 "copybara-service[bot] avatar"){kind=avatar}
I noticed you guys don't have a `.gitignore` yet so I added one here. This one combines the following github-provided `.gitignore` files: - https://github.com/github/gitignore/blob/main/Python.gitignore - https://github.com/github/gitignore/blob/main/community/Bazel.gitignore - https://github.com/github/gitignore/blob/main/Global/Patch.gitignore - https://github.com/github/gitignore/blob/main/Global/macOS.gitignore
Add logs in `consistent_restore_mesh` to examine the invariant.
Need to add because parent CL updates Orbax codebase.