Machine Learning Researcher (AI Infrastructure)

TL;DR

Machine Learning Researcher (AI Infrastructure): Conduct applied ML research to make decentralised model training practical with an accent on training dynamics, distributed optimisation, and scaling laws. Focus on contributing to infrastructure and core ML problems, productionising ideas, and open-source contributions.

Location: California, London

Salary: 240,000 USD - 360,000 USD Base + Equity

Company

London-based AI startup democratising AI training infrastructure via personal devices compute, raised ~$20M since March 2024 launch, open-sourced project with 15k GitHub stars.

What you will do

• Conduct applied ML research for decentralised model training.
• Contribute to infrastructure and core ML problems including training dynamics, distributed optimisation, scaling laws.
• Collaborate with software engineers to productionise ideas and open-source contributions.

Requirements

• Strong background in ML research; recent postdocs or early-career researchers welcome.
• Experience working on model training (LLMs, diffusion models, or related).
• High levels of curiosity and bias toward experimentation and building.

Nice to have

• Competition experience (ICPC, IOI, IMO, IPhO).