TorchDynamo: An Experiment in Dynamic Python Bytecode Transformation

compiler
3

Support for training is planned and a key priority for TorchDynamo. It will be based on the (also experimental) compilation / AOT autograd work found in the functorch repo.