Research Scientist · Google

Ashwini Pokle

My research aims to make generative models more capable and efficient by developing algorithms and architectures that generate higher quality images with lower computational cost. I work across sampling, training, and model design to improve the fidelity, controllability, and alignment of generated media.

I received my PhD from the Machine Learning Department at Carnegie Mellon University, advised by Prof. Zico Kolter. My thesis explored efficient generative inference by combining Deep Equilibrium Models and Diffusion Models. Prior to that, I obtained my MS in Computer Science at Stanford University. I also spent time at Stanford Vision and Learning Lab, where I worked on behavioral and social robot navigation under supervision of Prof. Silvio Savarese. I earned my BE (Hons.) in Computer Science from BITS Pilani, India. I have also interned at FAIR (Meta AI) and Bosch AI.

Ashwini Pokle
Research

My research advances generative modeling for images. I'm interested in designing novel sampling strategies, model architectures, improving training pipelines, and developing methods that enhance the overall fidelity and alignment of generated media.

Diffusion Models Consistency Models Deep Equilibrium Models Flow Matching Inverse Problems
Publications
2026
Learn to guide your diffusion model
Alexandre Galashov, Ashwini Pokle, Arnaud Doucet, Arthur Gretton, Mauricio Delbracio, Valentin De Bortoli
International Conference on Learning Representations (ICLR), 2026
2025
Consistency Models Made Easy
Zhengyang Geng, Ashwini Pokle, William Luo, Justin Lin, Zico Kolter
International Conference on Learning Representations (ICLR), 2025
2024
Training-free Linear Image Inverses via Flows
Ashwini Pokle, Matthew J. Muckley, Ricky T. Q. Chen, Brian Karrer
Transactions on Machine Learning Research (TMLR), 2024
2023
Deep Equilibrium Based Neural Operators for Steady-State PDEs
Tanya Marwah*, Ashwini Pokle*, J. Zico Kolter, Zachary Chase Lipton, Jianfeng Lu, Andrej Risteski
Advances in Neural Information Processing Systems (NeurIPS), 2023
One-Step Diffusion Distillation via Deep Equilibrium Models
Zhengyang Geng*, Ashwini Pokle*, J. Zico Kolter
Advances in Neural Information Processing Systems (NeurIPS), 2023
2022
Deep Equilibrium Approaches to Diffusion Models
Ashwini Pokle, Zhengyang Geng, J. Zico Kolter
Advances in Neural Information Processing Systems (NeurIPS), 2022
Path independent equilibrium models can better exploit test-time computation
Cem Anil*, Ashwini Pokle*, Kaiqu Liang*, Johannes Treutlein, Yuhuai Wu, Shaojie Bai, J. Zico Kolter, Roger Baker Grosse
Advances in Neural Information Processing Systems (NeurIPS), 2022
Contrasting the Landscape of Contrastive and Non-Contrastive Learning
Ashwini Pokle*, Jinjin Tian*, Yuchen Li*, Andrej Risteski
Conference on Artificial Intelligence and Statistics (AISTATS), 2022
2019 – 2021
Visually-Grounded Library of Behaviors for Generalizing Manipulation Across Objects, Configurations and Views
Hsiao-Yu Tung*, Jingyun Yang*, Yunchu Zhang*, Gaurav Pathak, Ashwini Pokle, Christopher G. Atkeson, Katerina Fragkiadaki
Conference on Robot Learning (CoRL), 2021
Deep Local Trajectory Planning and Control for Robot Navigation
Ashwini Pokle, Roberto Martín-Martín, Patrick Goebel, Patrick Goebel, Vincent Chow, Hans M Ewald, Junwei Yang, Zhenkai Wang, Amir Sadeghian, Dorsa Sadigh, Silvio Savarese, Marynel Vázquez
IEEE International Conference on Robotics and Automation (ICRA), 2019
Translating Navigation Instructions in Natural Language to a High-Level Plan for Behavioral Robot Navigation
Xiaoxue Zang*, Ashwini Pokle*, Marynel Vázquez, Kevin Chen, Juan Carlos Niebles, Alvaro Soto, Silvio Savarese
Conference on Empirical Methods in Natural Language Processing (EMNLP), 2018

Outside of research, I find balance in things that are wonderfully slow — tending a garden and making art by hand.

Gardening
Growing things slowly

I grow herbs, vegetables, and flowers while learning that not everything can be optimized. Gardening keeps me patient and connected to seasons 🤗.

Art
Making things by hand

I make art as a way to think differently while playing with colors without the pressure of an objective function.