GitHub repository: lalalland/gradient-checkpointing
Primary language: Python
Description: Make huge neural nets fit in memory