Founding Machine Learning Engineer

++++

Please apply to this role on our ATS

https://jobs.gem.com/atla-ai-com/am9icG9zdDqx_SrCygK6boqHZ3REZn5l

++++

As Atla’s 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 Atla’s 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 Atla team.

Please note that this role is in-person (we can sponsor visas and offer international relocation support!)

*This role is supported by AI Futures Grants, a UK Government scheme designed to help the next generation of AI leaders meet the costs of relocating to the UK. AI Futures Grants provide financial support to reimburse relocation costs such as visa fees, the immigration health surcharge and travel/subsistence expenses. Successful candidates for this role may be able to get up to £10,000 to meet relocations costs, subject to terms and conditions.

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 at Atla 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.

Compensation

• Competitive salary
• Significant stake in equity as one of the first joiners

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