flax
Forward experimental checkpoint policy option in lift.remat
#1560
Merged

Commits
  • Forward experimental checkpoint policy option in lift.remat
    jheek committed 4 years ago
Loading