Cole Harrison

Cole Harrison, machine learning research engineer

I am a Machine Learning Research Engineer at Amazon Web Services and Embodied AI lead at Cohere Labs, and collaborate with PRIOR at Ai2 and the Stanford Geometric Computation Group.

My research interests are in reinforcement learning, computer vision, multimodal foundation models, and scalable approaches for robot learning. My experience ranges from algorithm development to large-scale training and deployment.

Papers

  1. Yang You, Yi Du, Cole Harrison, Leonidas Guibas

    9D pose axes overlaid on unseen objects
  2. Le Qiu, Cole Harrison, Jiankai Sun, Yao Liu, Suning Huang, Qianzhong Chen, Yang You, Marco Pavone

  3. Haoquan Fang, Jiafei Duan, … Cole Harrison, … Ali Farhadi, Dieter Fox, Ranjay Krishna (29 authors)

  4. Shirui Chen, Cole Harrison, Ying-Chun Lee, Angela Jin Yang, Zhongzheng Ren, Lillian J. Ratliff, Jiafei Duan, Dieter Fox, Ranjay Krishna

News

Code

  • TOPReward

    Reference implementation and evaluation tooling for extracting dense task-progress signals from pretrained video-language models.

  • MolmoAct 2

    Pretraining data-quality filtering using TOPReward thresholding, and value-head work, for an open action-reasoning model family released with models, data, and evaluation rollouts.

  • UniPose9D

    Implementation and pretrained models for category-agnostic 9D object pose estimation from RGB and RGB-D input, with no mesh, category label, or reference view required.

  • LeRobot

    TOPReward and MolmoAct 2 integrations into Hugging Face's robotics library, plus fixes for Koch 1.1 hardware and setup failure modes.

  • MeGPT

    A QLoRA fine-tuning pipeline for training a personalized Llama-based chatbot from private message history.