Michael Psenka

Michael Psenka

Founding Principal Research Scientist · Base Labs

I lead Base Labs, an open-source AI research lab in Baseten.

I lead Base Labs, an AI research lab in Baseten that is fully open source and publishing, with the goal of advancing and democratizing engineered intelligence. I got my PhD from UC Berkeley EECS / BAIR, where I was advised by Aditi Krishnapriyan and worked closely with Yann LeCun, Pieter Abbeel, and Yi Ma. My dissertation, titled "Variational Approaches to Path Problems in Deep Learning", integrates dynamical systems theory with modern deep learning in various domains, including: reinforcement learning, generative molecular dynamics, and world models. Before Berkeley, I studied pure math at Princeton.

Education

UC Berkeley — PhD in EECS, 2021–2026
Princeton University — A.B. in Mathematics, 2017–2021

News

Publications

Parallel Stochastic Gradient-Based Planning for World Models
Michael Psenka, Michael Rabbat, Aditi Krishnapriyan, Yann LeCun*, Amir Bar*
ICML 2026
Action-Minimization Meets Generative Modeling: Efficient Transition Path Sampling with the Onsager-Machlup Functional
Sanjeev Raja*, Martin Šípka*, Michael Psenka*, Tobias Kreiman, Michal Pavelka, Aditi S. Krishnapriyan
ICML, July 2025
Learning a Diffusion Model Policy from Rewards via Q-Score Matching
Michael Psenka*, Alejandro Escontrela*, Pieter Abbeel, Yi Ma
ICML, July 2024
Representation Learning via Manifold Flattening and Reconstruction
Michael Psenka, Druv Pai, Vishal Raman, Shankar Sastry, Yi Ma
JMLR, May 2024
Role of Uncertainty in Anticipatory Trajectory Prediction for a Ping-Pong Playing Robot
Nima Rahmanian, Michael Gupta, Renzo Soatto, Srisai Nachuri, Michael Psenka, Yi Ma, Shankar Sastry
arXiv, Nov. 2023
Pursuit of a Discriminative Representation for Multiple Subspaces via Sequential Games
Druv Pai, Michael Psenka, Chih-Yuan Chiu, Manxi Wu, Edgar Dobriban, Yi Ma
Journal of the Franklin Institute, April 2023
A Proof of The Triangular Ashbaugh-Benguria-Payne-Pólya-Weinberger Inequality
Ryan Arbon*, Mohammed Mannan*, Michael Psenka*, Seyoon Ragavan*
Journal of Spectral Theory, Sept. 2022
CTRL: Closed-Loop Transcription to an LDR via Minimaxing Rate Reduction
Xili Dai, Shengbang Tong, Mingyang Li, Ziyang Wu, Michael Psenka, Kwan Ho Ryan Chan, Pengyuan Zhai, Yaodong Yu, Xiaojun Yuan, Heung Yeung Shum, Yi Ma
Entropy Journal, March 2022
Second-Order Optimization for Tensors with Fixed Tensor-Train Rank
Michael Psenka, Nicolas Boumal
NeurIPS OPT 2020
Reconstruction Without Registration
Michael Psenka, Tolga Birdal, Leonidas Guibas
IROS Geometric Methods Workshop 2020