flax
93d3eb78 - support transform reuse for lift.jit decorator

Commit

2 years ago

support transform reuse for lift.jit decorator PiperOrigin-RevId: 599544350

References

Author

Cristian Garcia

Committer

a-googler

a-googler

Parents

Loading