Software Engineer, Machine Learning

Imbue; United States; Remote; USD 140,000 - 350,000; Full time

Job summary

A leading company in AI research seeks a Machine Learning Engineer to work on deep learning projects. You'll collaborate with a senior researcher to create human-like intelligence systems, implementing innovative solutions and optimizing performance. This role offers flexibility in work location and a competitive compensation package, encouraging applicants to showcase their relevant work.

Benefits

Flexible working hours

Time and budget for learning and self-improvement

Qualifications

• Comfortable writing Python.
• Familiar with PyTorch and training deep neural networks.

Responsibilities

• Implement papers to improve code generation and debugging.
• Analyze failure modes and benefits of techniques.
• Contribute to data generation pipeline.

Skills

Python

PyTorch

Job description

Summary

As a machine learning engineer, you’ll work 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

1. 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.
2. Devise scientifically rigorous experiments to analyze the failure modes and benefits of different techniques.
3. Contribute to our data generation pipeline for creating instruction-tuning and evaluation data.
4. Develop infrastructure for making fine-tuning and experimentation simpler, faster, and less error-prone.
5. Work with our data and product engineers to identify areas of improvement from model results.
6. 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. The salary range for this role is $140,000–$350,000 in cash, and $10,000–$5,000,000 in equity. We encourage you to apply even if your requirements fall outside this range.

How to apply

Include notes on why you're interested in working with us, and showcase any relevant work (open source code, side projects, etc.). We aim to reply within a week or two at most.

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 work safely in the real world. We train our own foundation models optimized for reasoning and prototype agents on top of these models. 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 what we love.