Dynamic Tensor Rematerialization

#1

请问MegEngine是否有向Pytorch提供DTR相关的pull request? 之前在知乎上看到过相关信息,或者说MegEngine是否有考虑开放DTR相关拓展,使用户能在pytorch框架内无缝调用MegEngine针对activation checkpointing的自动优化功能?

#2

没有。请关注 DTR 原作者的 PR https://github.com/pytorch/pytorch/pull/42056