Book
Open access
Jul 2026
Optimus: A Generic Operator-Level PyTorch Model Transformation Framework
Optimus is a general-purpose model transformation framework built in the PyTorch 2.x (PT2) machine learning compiler that applies an efficient greedy search algorithm for pattern matching and replacement, while preserving model semantic.
Menglu Yu, Jiaqi Xu, Yuzhen Huang et al.
· Proceedings of the 32nd ACM... · 0 citations