Machine Learning Engineer

AI research and deployment company training frontier models to evaluate generative AI and enable safe development of artificial general intelligence.
$125,000 - $375,000
Machine Learning
Senior Software Engineer
In-Person
11 - 50 Employees
3+ years of experience
AI

Description For Machine Learning Engineer

Atla is an innovative AI research and deployment company focused on enabling the safe development of artificial general intelligence. As a Machine Learning Engineer, you'll play a crucial role in spearheading post-training and inference frameworks for large language models. The position offers an opportunity to work with cutting-edge AI technology while contributing to the safe development of AI systems.

The role involves designing scalable post-training frameworks, implementing efficient data pipelines, and engineering robust inference platforms. You'll collaborate with researchers and engineers to accelerate the iteration of new research ideas and improve training workflows. The ideal candidate should have strong expertise in ML engineering, experience with orchestration systems, and proficiency in serving large machine learning models at scale.

Working at Atla means joining a team of researchers, engineers, and operational leaders backed by Y Combinator, Creandum, and prominent founders. The company offers exceptional compensation, including significant equity, comprehensive benefits, and the chance to make a meaningful impact in AI safety. This in-person role in London comes with visa sponsorship and relocation support through the UK AI Futures partnership.

The position requires 3+ years of experience and offers a competitive salary range of £100K - £300K GBP. You'll be working in a collaborative, fast-paced startup environment where continuous learning and adaptation are essential. Join Atla to help shape the future of safe AI development while working with state-of-the-art technology and a dedicated team.

Last updated 3 hours ago

Responsibilities For Machine Learning Engineer

  • Design and optimise scalable post-training framework for managing training runs
  • Implement scalable data pipelines and optimize models for performance
  • Engineer robust, high-performance inference platforms
  • Collaborate with researchers to accelerate iteration of research ideas
  • Build and grow engineering organisation

Requirements For Machine Learning Engineer

Python
Kubernetes
  • Proven expertise in software or ML engineering
  • Experience with orchestration systems like SLURM or Ray
  • Skilled in creating multi-instance clusters for data and model parallel training
  • Proficient in serving large machine learning models at scale
  • Strong understanding of techniques like paged attention and gradient checkpointing

Benefits For Machine Learning Engineer

Medical Insurance
Dental Insurance
Vision Insurance
401k
Equity
Visa Sponsorship
  • Exceptionally competitive salary
  • Significant equity stake
  • Pension plan
  • Medical, dental, and vision benefits
  • Visa sponsorship
  • International relocation support

Interested in this job?

Jobs Related To Atla Machine Learning Engineer

Research Engineer

Senior Research Engineer position at Atla developing language models and safety guardrails for frontier AI systems

AI app/SDK test and automation development - Sr. Engineer

Senior Engineer position at Qualcomm China focusing on AI app/SDK testing and automation development, requiring 3+ years experience and expertise in mobile testing and AI frameworks.

Software Engineer - ML Infrastructure

Senior ML Infrastructure Engineer role at Salesforce, building scalable AI services and working with cutting-edge ML technologies.

Senior Machine Learning Engineer - Computer Vision

Senior Machine Learning Engineer position at DoorDash focusing on Computer Vision development for retail and grocery delivery optimization

Machine Learning Engineer

Senior Machine Learning Engineer role at DoorDash focusing on ETA prediction and routing optimization for delivery logistics.