| Name |
Last commit
|
Last update |
|---|---|---|
| .. | ||
| __init__.py | ||
| _make.py | ||
| annotation.py |
* add checkpoint annotation for checkpointing memory optimization * add alpha-equivalence checkpoint test and fix gradient type issue * fix build issues * ignore checkpoint annotation when checking missing gradients * refactor, fix checkpoint compute for tuple and add tests
| Name |
Last commit
|
Last update |
|---|---|---|
| .. | ||
| __init__.py | Loading commit data... | |
| _make.py | Loading commit data... | |
| annotation.py | Loading commit data... |