X-cache: a modular architecture for domain-specific caches. - Ali Sedaghati; Milad Hakimi; Reza Hojabr; Arrvindh Shriraman (ISCA, 2022)

TapeFlow: Streaming Gradient Tapes in Automatic Differentiation. - Milad Hakimi; Arrvindh Shriraman (CGO, 2024)