Pattern Matching in AI Compilers and its Formalization (Extended Version)

Joseph W. Cutler, Alex Collins, Bin Fan +2

PyPM is a Python-based domain specific language (DSL) for building rewrite-based optimization passes on machine learning computation graphs. Users define individual optimizations b…