We're planting a tree for every job application! Click here to learn more

Machine Learning Research Engineer


San Francisco, CA, US

Posted 17 days ago

ML Research Engineer at AI Research Lab, Imbue

Tech stack

  • PyTorch

This is an on-site position


In this role, you’ll work with Imbue's researchers to do cutting-edge deep learning research—conducting experiments, creating infrastructure, and developing tooling & visualizations—with the goal of developing more human-like machine intelligence.

Note: This research role requires being onsite in San Francisco. If you're remote, please take a look at the Machine Learning Engineer (Remote) role, which is more engineering-heavy.

Example projects:

  • Implement a self-supervised network using contrastive and reconstruction losses.
  • Create a library on top of PyTorch to enable efficient network architecture search.
  • Implement networks from newly published papers.
  • Run massively parallel experiments to understand all variants of an architecture.
  • Develop more realistic simulations for training Imbue's agents.
  • Create visualizations to help deeply understand what Imbue's networks learn and why.

You are:

  • Passionate about understanding the fundamentals of intelligence.
  • Very comfortable writing Python.
  • Excited to be a world-class ML engineer.
  • A fan of paired programming.
  • Passionate about engineering best practices.


  • Work directly on creating software with human-like intelligence.
  • Generous compensation, equity, and benefits.
  • $20K+ yearly budget for self-improvement: coaching, courses, conferences, etc.
  • Actively co-create and participate in a positive, intentional team culture.
  • Spend time learning, reading papers, and deeply understanding prior work.
  • Frequent team events, dinners, off-sites, and hanging out.

About the company:

Imbue builds AI systems that reason and code, enabling AI agents to accomplish larger goals and safely work in the real world. Imbue trains its own foundation models optimized for reasoning and prototype agents on top of these models. By using these agents extensively, they can gain insights into improving both the capabilities of the underlying models and the interaction design for agents.

Imbue aims to rekindle the dream of the *personal* computer, where computers become truly intelligent tools that empower us, giving us freedom, dignity, and agency to pursue the things we love.

WorksHub are not directly affiliated/are not a direct part of Imbue

What makes you a perfect
candidate for this role

  • An academic degree in the relevant field is good to have

  • 4+

    years of commercial experience
  • Corresponding level of skills:



    Machine Learning


  • Language skills:





Role type

Full time

Visa sponsorship


Benefits & perks

  • Flexible Working

  • Conferences

  • Equity

  • Supportive culture

  • Team events

  • Learning budget

Similar roles that might interest you


CareersCompaniesSitemapFunctional WorksBlockchain WorksJavaScript WorksAI WorksGolang WorksJava WorksPython WorksRemote Works

Ground Floor, Verse Building, 18 Brunswick Place, London, N1 6DZ

108 E 16th Street, New York, NY 10003

Subscribe to our newsletter

Join over 111,000 others and get access to exclusive content, job opportunities and more!

© 2024 WorksHub

Privacy PolicyDeveloped by WorksHub