flax
76dec35f - Add more explicit warnings and logs for code with Orbax backend.

Commit

2 years ago

Add more explicit warnings and logs for code with Orbax backend. PiperOrigin-RevId: 524383695

References

Author

IvyZX

Committer

a-googler

Parents