Requirements

• Bachelor's degree in Computer Science, or related technical field, or equivalent practical experience.
• 2 years of software engineering experience with Python-based ML/scientific libraries such as JAX, or internship experience/Open Source Software contributions in these areas.
• 1 year of experience with machine learning.

Nice to Haves

• Experience with large-scale system design, distributed systems, or performance engineering.
• Good understanding of engineering best practices, codebase design, and testing.
• Familiarity with training of modern large language models.
• Ability to collaborate and communicate effectively with diverse teams across time zones.

What You'll Be Doing

• Working on long-term projects to enable new research involving multiple team members, requiring solid software engineering and design skills.
• Responding to time-sensitive issues during large-scale training runs and ongoing research.
• Making systematic improvements to technical workflows and tooling used in pretraining.
• Applying engineering expertise to help researchers tackle technical challenges, understanding both codebase and broader technical stack, model architecture, and data pipeline.
• Balancing good engineering practices with maintaining a flexible codebase that researchers find easy to use and extend.

Perks and Benefits

• Competitive pay ranging approximately from $147,000 to $210,000 USD, plus 15% bonus target, equity, and benefits.
• Opportunities for diverse learning experiences and career pathways in a pioneering AI lab.
• Work at locations in London, UK or Mountain View, CA, USA with preferred working location options.
• Engagement with cutting-edge AI research and deployment at scale for global impact.