/FastCkpt

Python package for rematerialization-aware gradient checkpointing

Primary LanguagePythonApache License 2.0Apache-2.0