Software Engineer, Machine Learning

Sorry, this job was removed at 08:13 p.m. (CST) on Wednesday, Jun 04, 2025
2 Locations
Remote or Hybrid
140K-350K Annually
Artificial Intelligence • Machine Learning • Other
The Role
Summary
As a machine learning engineer, you’ll work very closely with a senior member of our research team on cutting-edge deep learning research, infrastructure, and tooling towards the goal of creating general human-like machine intelligence.

This role can be either fully in-person or remote.


Example projects
• Understand and implement papers to improve areas of interest, such as automatically generating and iteratively refining code, outputting reasoning traces, and verifying or debugging code.
• Devise scientifically rigorous experiments to analyze the failure modes and benefits of different techniques.
• Contribute to our data generation pipeline for creating instruction-tuning and evaluation data
• Develop infrastructure for making fine-tuning and experimentation simpler, faster, and less error-prone.
• Work with our data and product engineers to identify areas of improvement from model results
• Leverage our existing framework to perform hyperparameter sweeps, and maintain and contribute new features to the sweep code


You are
• Very comfortable writing Python.
• Familiar with PyTorch and training deep neural networks.
• Passionate about engineering best practices.
• Self-directed and independent.
• Excellent at getting things done.


Compensation and Benefits
Work directly on creating software with human-like intelligence
Flexible working hours
Time and budget for learning and self improvement
Compensation packages are highly variable based on a variety of factors. If your salary requirements fall outside of the stated range, we still encourage you to apply. The range for this role is $140,000–$350,000 cash, $10,000–$5,000,000 in equity.


How to apply

All submissions are reviewed by a person, so we encourage you to include notes on why you're interested in working with us. If you have any other work that you can showcase (open source code, side projects, etc.), certainly include it! We know that talent comes from many backgrounds, and we aim to build a team with diverse skillsets that spike strongly in different areas.

We try to reply either way within a week or two at most (usually much sooner).

Learn more about our full interview process here.


About us

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

We aim 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.

Similar Jobs

Upstart Logo Upstart

Senior Software Engineer

Artificial Intelligence • Fintech • Machine Learning • Social Impact • Software
Easy Apply
Remote
United States
1500 Employees
164K-226K Annually

Socure Logo Socure

Senior Software Engineer

Artificial Intelligence • Machine Learning • Software • Analytics
Remote
USA
386 Employees
180K-200K Annually

MARA Logo MARA

Lead Software Engineer

Big Data • Information Technology
Easy Apply
Remote
USA
180 Employees

SmarterDx Logo SmarterDx

Senior Software Engineer

Artificial Intelligence • Healthtech • Software
Easy Apply
Remote
United States
310 Employees
190K-230K Annually
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: San Francisco, CA
15 Employees
Year Founded: 2021

What We Do

We build AI systems that can reason, in order to enable AI agents that can accomplish larger goals and safely work for us in the real world. To do this, we train foundation models optimized for reasoning. On top of our models, we prototype agents to accelerate our own work, seriously using them in order to shed light on how to improve the underlying model capabilities, as well as the interaction design for agents.

We aim to rekindle the dream of the *personal* computer—for computers to be truly intelligent tools that empower us, giving us freedom, dignity, and agency to do the things we love.

Gallery

Gallery

Similar Companies Hiring

Scotch Thumbnail
Software • Retail • Payments • Fintech • eCommerce • Artificial Intelligence • Analytics
US
25 Employees
Milestone Systems Thumbnail
Software • Security • Other • Big Data Analytics • Artificial Intelligence • Analytics
Lake Oswego, OR
1500 Employees
Idler Thumbnail
Artificial Intelligence
San Francisco, California
6 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account