Email: [email protected]

Publications

• Towards Learning High-Precision Least Squares Algorithms with Sequence Models.
  Jerry Liu, Jessica Grogan, Owen Dugan, Ashish Rao, Simran Arora, Atri Rudra, Chris Ré In ICLR: The Thirteenth International Conference on Learning Representations, April. 2025. Poster presentation
  [Paper] [Code]
• Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture
  Daniel Y. Fu, Simran Arora*, Jessica Grogan*, Isys Johnson*, Sabri Eyuboglu*, Armin W. Thomas*, Benjamin F. Spector, Michael Poli, Atri Rudra, Christopher Ré In NeurIPS: 27th Conference on Neural Information Processing Systems , Dec. 2023. Oral presentation
  [Paper] [Code] [Blog Post]

• Monarch: Expressive Structured Matrices for Efficient and Accurate Training
  Tri Dao, Beidi Chen, Nimit Sohoni, Arjun Desai, Michael Poli, Jessica Grogan, Alexander Liu, Aniruddh Rao, Atri Rudra, Christopher Ré.
  In ICML: 39th International Conference on Machine Learning, July 2022. Outstanding Paper Runner-up
  [Paper] [Code] [Poster] Outstanding Poster Award at the EfficientML Bay Area meetup