flax
[nnx] refactor remat
#4662
Open

[nnx] refactor remat #4662

cgarciae wants to merge 1 commit into main from improve-remat-docs
cgarciae
IvyZX
IvyZX approved these changes on 2025-03-28
IvyZX IvyZX added pull ready
cgarciae [nnx] refactor remat
c4145b60
cgarciae cgarciae force pushed from 22e3417c to c4145b60 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone