ryamabra/ckptplan-framework: Python library for memory-constrained activation checkpoint optimization, recomputation scheduling, and GPU training performance analysis.

Python library for memory-constrained activation checkpoint optimization, recomputation scheduling, and GPU training performance analysis.

Read Original

Related