Machine Learning Infrastructure Engineer

We are committed to engineering safe, beneficial AI systems that will have a massive positive impact on the future of humanity. We are a London-based start-up building the next-generation of helpful and honest AI. Become part of our growing world-class team, backed by Y Combinator, Creandum, and the founders of Reddit, Cruise, Rappi, Instacart and more.

Role

As a Founding Machine Learning Engineer, you’ll spearhead the scaling of our fine-tuning and inference frameworks for large language models.

• Build an automated fine-tuning framework that can scale to the next order of magnitude of training runs
• Explore novel synthetic data generation techniques in close collaboration with our research team to ensure high-quality and diverse data at scale.
• Engineer robust, high-performance inference platforms to ensure our products’ reliability, throughput and speed at scale.
• Collaborate in defining and steering evolving inference and training stack working with researchers, founders and advisors to develop the most powerful evaluation models.
• Found and grow our engineering organisation, setting a high bar of excellence as a core member of the team.

Qualifications

Evidence of exceptional ML engineering ability:

• Have experience operating orchestration systems such as SLURM, Ray, or similar.
• Experience in creating and managing multi-instance clusters for data and model parallel training across GPUs/TPUs, preferably using PyTorch and Kubernetes.
• Proficiency in efficient serving of large machine learning models at scale, including quantisation and distributed computing.
• Understanding around the latest AI research and a familiarity of the knowledge of how these systems can be efficiently implemented.

Nice to have

• Experience at a leading machine learning company (Mistral, Anthropic, X.ai, HuggingFace, Neuralink, OpenAI, etc.)
• Interested in and thoughtful about of the impacts of AI technology.

About you
You are going to thrive with the following mindset:

• Collaborative and team-oriented, with strong communication skills.
• Comfortable with the uncertainty and fast pace of a hyper-growth startup.
• Willingness to continuously learn and adapt in a dynamic environment.
• Unpretentious and hard working; find the best ideas wherever they come from.

Join our driven founding team to make a dent in the universe with safe, beneficial AI systems!