As a Machine Learning Engineer, you'll implement deep learning research, develop infrastructure, and enhance data processing to create advanced AI systems.
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.
Top Skills
Python
PyTorch
Similar Jobs
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
As a Staff AI/ML Software Engineer, you'll design ML models, build data pipelines, and create scalable systems that enhance user experience through search ranking and personalization.
Top Skills:
AWSAzureGCPJavaKafkaKubernetesLightgbmMl FrameworksPythonPyTorchScalaSparkSpring BootTensorFlowXgboost
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
The Staff AI/ML Software Engineer designs scalable ML systems, builds data pipelines, and optimizes algorithms for personalization, search ranking, and ML models while ensuring high performance and governance.
Top Skills:
JavaKafkaKubernetesLightgbmPythonPyTorchScalaSparkSpring BootTensorFlowXgboost
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Build scalable AI/ML systems for search ranking and personalization, develop real-time ML models, and optimize algorithms. Collaborate across teams to enhance customer experiences through AI-driven technology.
Top Skills:
JavaKafkaKubernetesLightgbmPythonPyTorchScalaSparkSpring BootTensorFlowXgboost
What you need to know about the Boston Tech Scene
Boston is a powerhouse for technology innovation thanks to world-class research universities like MIT and Harvard and a robust pipeline of venture capital investment. Host to the first telephone call and one of the first general-purpose computers ever put into use, Boston is now a hub for biotechnology, robotics and artificial intelligence — though it’s also home to several B2B software giants. So it’s no surprise that the city consistently ranks among the greatest startup ecosystems in the world.
Key Facts About Boston Tech
- Number of Tech Workers: 269,000; 9.4% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Thermo Fisher Scientific, Toast, Klaviyo, HubSpot, DraftKings
- Key Industries: Artificial intelligence, biotechnology, robotics, software, aerospace
- Funding Landscape: $15.7 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Summit Partners, Volition Capital, Bain Capital Ventures, MassVentures, Highland Capital Partners
- Research Centers and Universities: MIT, Harvard University, Boston College, Tufts University, Boston University, Northeastern University, Smithsonian Astrophysical Observatory, National Bureau of Economic Research, Broad Institute, Lowell Center for Space Science & Technology, National Emerging Infectious Diseases Laboratories
