Dynamo
ThunderCompiler— A class that compiles a torch.fx.GraphModule to a thunder.ThunderModule. This class is meant to be used as a backend for the torch.compile() function.thunderfx— Compiles a callable (function or model) by using Thunder as the backend of torch.compile().ThunderFXCompiledObject— A wrapper around the result of thunderfx() compilation.
Was this page helpful?
Thanks for the feedback.