torch.compile
Named by 10 kept items in this workspace.
8 items name it under frameworks and tools and 2 under classification.technologies.
Search every workspace for torch.compile. It matches the name as written, so it can find other things that share it.
The items that name it 10
Model compilation combined with KV caching to make text generation more efficient.
Compiles the model for Trainium during generation with a static cache.
With backend='neuron', TorchDynamo captures bytecode as an FX graph, AOTAutograd lowers it to ATen operations, and the Neuron backend fuses them into a single NEFF run as one hardware dispatch.
Compilation for better performance on TPU, using backend="tpu".
Used for the MoE decoding benchmarks. The team is working to remove graph breaks so more models compile end to end.
Compiler cited as limited to transformations written in advance by engineers, in contrast with Maxwell.
PyTorch graph capture; in TorchTPU uses TorchDynamo and AOTAutograd to produce an FX graph lowered to StableHLO.
PyTorch JIT compilation, enabled/removed selectively across model backends to control latency and memory
All technologies in this workspace.