Search Results for author: Mike He

Found 1 papers, 1 papers with code

Dynamic Tensor Rematerialization

2 code implementations ICLR 2021 Marisa Kirisame, Steven Lyubomirsky, Altan Haan, Jennifer Brennan, Mike He, Jared Roesch, Tianqi Chen, Zachary Tatlock

Checkpointing enables the training of deep learning models under restricted memory budgets by freeing intermediate activations from memory and recomputing them on demand.

Cannot find the paper you are looking for? You can Submit a new open access paper.