flax
70d6840c - Merge pull request #1361 from andsteing:optax2

Commit

4 years ago

Merge pull request #1361 from andsteing:optax2 PiperOrigin-RevId: 379476057

References

#1361 - Updates examples/vae to use Optax.

Author

a-googler

a-googler

Parents

Loading