Research

RL tree search over decoding policies for discrete diffusion models. done as a test-time scaling axis for gemini diffusion. [paper] | [code] † *

Variance-aware inference scaling methods for robotics diffusion models. † *

Theoretical framework and proof-of-concept for trajectory-search inference scaling of stochastic diffusion models, expanding on deepmind's work for deterministic models. [paper] | [code] *

Diagnostics for brain tumors using medical images and computer vision. Generative models for dataset balancing. I learned a lot about building infra for running lots of experiments in parallel. †