Rohit Dilip

I research generative modeling for biology at Caltech. Recently, I've worked on using flow matching to develop scalable tokenizers and generating new biological structures using autoregressive models. Before grad school, I studied quantum information at TUM and got my bachelor's degree in physics from Princeton. I'm also a co-organizer of Project Synapse.

Rohit Dilip
Ghibli-style portrait

Projects

picomap Easy dataloading for machine learning

Publications

Adaptive Protein Tokenization (ICML 2026)
Rohit Dilip, Evan Zhang, Ayush Varshney, David Van Valen
arXiv GitHub
We develop a new way to represent proteins where each token provides global information about the protein structure.
Kanzi: Flow Autoencoders are Effective Protein Tokenizers (ICLR 2026)
Rohit Dilip, Evan Zhang, Ayush Varshney, David Van Valen
arXiv bioRxiv GitHub
We develop a flow matching based autoencoder for protein structure tokenization. Kanzi outperforms 20x larger models trained on 400x as much data.
Jovian: Bidirectional Autoregressive Protein Structure Generation
Rohit Dilip, Ayush Varshney, Evan Zhang, David Van Valen
Jovian presents an autoregressive approach to condition generative models on particular substructures.
CellSAM: A Foundation Model for Cell Segmentation
Uriah Israel, Markus Marks, Rohit Dilip, Qilin Li, et al.
arXiv GitHub
CellSAM is a generalist model for cell segmentation that works across visual modalities, sizes, and shapes.
Data compression for quantum machine learning
Rohit Dilip, Leo Li, Adam Smith, Frank Pollmann
Physical Review Research, 2023
arXiv GitHub
We develop a tensor-network based data compression technique specifically for exploitation on low-depth quantum circuits.
Real- and imaginary-time evolution with compressed quantum circuits
ShengHsuan Lin, Rohit Dilip, Adam Smith, Frank Pollmann
PRX Quantum
arXiv
We show that low depth classical simulations of quantum circuits can be used to simulate non-equilibrium quantum dynamics.