yalechang/gradient-checkpointing

Make huge neural nets fit in memory

Python