I’m a post-doctoral researcher in computer science at the University of Washington.

My current focus is egg, a toolkit for program optimization and synthesis powered by e-graphs and equality saturation. We’ve applied it to some really fun domains like improving floating point accuracy, shrinking 3D models, and linear algebra optimization.

During my PhD (also at UW), I worked mostly in programming languages (PLSE group), but I enjoyed collaborating with friends in molecular systems (MISL), and machine learning systems (SAMPL). I also worked on programming models for microfluidic chips. The Puddle framework aims to provide a safer, easier way to program these devices, which promise to make labs-on-a-chip (LoCs) scalable and affordable.

News

• 2021 Oct
  • Our paper on inferring rewrites using equality saturation was a Distinguished Paper at OOPSLA 2021.
• 2021 Sep
  • Our paper connecting e-matching to relational databases was (conditionally) accepted to POPL 2022.
• 2021 May
  • I defended my PhD! You can find my thesis here. I’ll be staying on as a postdoc to work on egg and related projects.
• 2021 Apr
  • I wrote a post on the SIGPLAN blog about egg.
• 2021 Jan
  • Our paper applying egg to deep learning graph rewriting will appear at MLSys 2021.
• 2020 Oct
  • Our paper on egg will appear at POPL 2021 as a “Distinguished Paper”!
    Check out the 5 min or 30 min video talks.
• 2020 Jun
  • I’ve been working on this e-graph library called egg for a while. I and my fellow egg-heads put out a paper describing the system and some of its novel ideas.
  • I’m doing a summer internship at OctoML using egg to optimize parts of the TVM compiler stack.
• 2020 Mar
  • Some friends and I wrote a fun paper about synthesizing 3D CAD models. It will appear (virtually!) at PLDI 2020.

Papers

• Relational E-Matching

  • Yihong Zhang
  • Remy Wang
  • Max Willsey
  • Zach Tatlock
  POPL 2022, January 2022

• Rewrite Rule Inference Using Equality Saturation

  • Chandrakana Nandi
  • Max Willsey
  • Amy Zhu
  • Remy Wang
  • Brett Saiki
  • Adam Anderson
  • Adriana Schulz
  • Dan Grossman
  • Zach Tatlock
  OOPSLA 2021, August 2021
  Distinguished Paper

• Equality Saturation for Tensor Graph Superoptimization

  • Yichen Yang
  • Phitchaya Mangpo Phothilimthana
  • Remy Wang
  • Max Willsey
  • Sudip Roy
  • Jacques Pienaar
  MLSys 2021, January 2021

• egg: Fast and Extensible E-graphs

  • Max Willsey
  • Remy Wang
  • Oliver Flatt
  • Chandrakana Nandi
  • Zach Tatlock
  • Pavel Panchekha
  POPL 2021, January 2021
  Distinguished Paper

• PurpleDrop: A Digital Microfluidics-based Platform for Hybrid Molecular-Electronics Applications

  • Ashley Stephenson
  • Max Willsey
  • Jeff McBride
  • Sharon Newman
  • Bichlien Nguyen
  • Chris Takahashi
  • Karin Strauss
  • Luis Ceze
  IEEE Micro 2020, July 2020

• Synthesizing Structured CAD Models with Equality Saturation and Inverse Transformations

  • Chandrakana Nandi
  • Max Willsey
  • Adam Anderson
  • James Wilcox
  • Eva Darulova
  • Dan Grossman
  • Zach Tatlock
  PLDI 2020, March 2020

• Scaling Microfluidics to Complex, Dynamic Protocols

  • Max Willsey
  • Ashley Stephenson
  • Chris Takahashi
  • Bichlien Nguyen
  • Karin Strauss
  • Luis Ceze
  ICCAD 2019, November 2019
  Invited Paper

• High Density DNA Data Storage Library via Dehydration with Digital Microfluidic Retrieval

  • Sharon Newman
  • Ashley Stephenson
  • Max Willsey
  • Bichlien Nguyen
  • Chris Takahashi
  • Karin Strauss
  • Luis Ceze
  Nature Communications, April 2019

• Mega-Microfluidics

  • Max Willsey
  • Luis Ceze
  WACI at ASPLOS 2019, April 2019

• Puddle: A Dynamic, Error-Correcting, Full-Stack Microfluidics Platform

  • Max Willsey
  • Ashley Stephenson
  • Chris Takahashi
  • Pranav Vaid
  • Bichlien Nguyen
  • Michal Piszczek
  • Christine Betts
  • Sharon Newman
  • Sarang Joshi
  • Karin Strauss
  • Luis Ceze
  ASPLOS 2019, April 2019

• DNA Data Storage and Hybrid Molecular–Electronic Computing

  • Douglas Carmean
  • Luis Ceze
  • Georg Seelig
  • Callie Bee
  • Karin Strauss
  • Max Willsey
  Proceedings of the IEEE, January 2019
  Invited paper

• Iterative Search for Reconfigurable Accelerator Blocks with a Compiler in the Loop

  • Max Willsey
  • Vincent Lee
  • Alvin Cheung
  • Ras Bodik
  • Luis Ceze
  IEEE TCAD, October 2018

• Extensible Semantics for Fluidics

  • Max Willsey
  • Jared Roesch
  OBT at POPL, January 2018

• Design and Implementation of Concurrent C0

  • Max Willsey
  • Rokhini Prabhu
  • Frank Pfenning
  Fourth International Workshop on Linearity, June 2016

• Design and Implementation of Concurrent C0

  • Max Willsey
  Senior Honors Thesis, May 2016
  Designated as Exemplary Thesis by the senior thesis award committee