flax
aed27f67 - Add tree-mode support for nnx.remat

Commit

101 days ago

Add tree-mode support for nnx.remat PiperOrigin-RevId: 870178626

References

#5242 - Add tree-mode support for nnx.remat

Author

Cristian Garcia

Committer

a-googler

a-googler

Parents

Loading