[PyTorch] Add op-level activation offload opt-out API#3108

timmoon10 merged 12 commits into

NVIDIA:mainNVIDIA/TransformerEngine:mainfrom

lhb8125:codex/te-op-offload-controllhb8125/TransformerEngine:codex/te-op-offload-controlCopy head branch name to clipboard