Table of Contents
Shortcuts

thunder.transforms

MaterializationTransform(device, *, init)

Materialize a model that can fit on a device only after transforms applied.

ConstantFolding()

Apply Constant Folding to computation trace.

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. Read PyTorch Lightning's Privacy Policy.