Looking for an advisor for research in ML systems and systems for ML.
Worked at the intersection of computer vision and robotics to develop a robot action policy that generalizes to unseen tasks and environments. Using the DROID dataset — a large-scale collection of diverse robot task demonstrations — we modified the policy's diffusion architecture to improve generalization.
Key contributions included creating visualizations for gripper heuristics to extract meaningful waypoints from noisy human-operated trajectories, training and evaluating baseline models through the full experimental pipeline (configuring training runs, monitoring loss, and conducting physical robot experiments), and investigating distributed data parallelism across multiple GPUs to accelerate training.
I also worked on data collection and experimentation, which ran on a daily cycle: train the model overnight, evaluate the policy on the physical robot the next day across roughly 80 trials spanning diverse objects and tasks, then use those results to refine the model before the next training run.