Pure Tensor Program Rewriting via Access Patterns (Representation Pearl)

Gus Henry Smith, Andrew Liu, Steven Lyubomirsky +5

Tensor kernels in machine learning (ML) often correspond to pure mathematical expressions, making term rewriting an attractive strategy for optimization and mapping to specialized…