flax
517b7635 - Removes `flax.optim.dynamic_scale`.

Commit

3 years ago

Removes `flax.optim.dynamic_scale`. The module was copied to `flax.training` in #2375. PiperOrigin-RevId: 467042598

References

Author

andsteing

Committer

a-googler

Parents