flax
451890e9 - Update axes_scan to use extend.linear_util

Commit

2 years ago

Update axes_scan to use extend.linear_util PiperOrigin-RevId: 563652936

References

#3313 - Update axes_scan to use extend.linear_util

Author

a-googler

a-googler

Committer

a-googler

a-googler

Parents

Loading