Machine Learning Research Engineer

We are seeking a Machine Learning Research Engineer to join our team. You will design and run rigorous experiments to improve model capability, develop data recipes, and advance training science.

Responsibilities:

• Design and run experiments to measure gains in model capability
• Improve data quality and develop data recipes
• Define metrics for RL data and rewards
• Advance training science and develop post-training recipes
• Build evaluations and targeted tasks to guide investments
• Collaborate with engineers to scale training runs

Basic Qualifications:

• 3+ years of experience training large neural networks (1B+ parameters)
• Experience writing production-quality code in Python, PyTorch, JAX, Rust, or C++

Preferred Skills and Experience:

• Experience with RL algorithms and language models
• Strong background in RL and machine learning fundamentals
• Excellent programming skills in Python and PyTorch or JAX
• Experience with distributed training and debugging
• Track record of original work in ML

Additional Requirements:

• Must be able to work onsite in Palo Alto

Compensation and Benefits:

• Annual salary: $220,000 - $357,000 USD
• Comprehensive benefits package, including equity, medical, vision, and dental coverage, 401(k) plan, and more