Job Description
++++
Please apply to this role on our ATS
https://jobs.gem.com/atla-ai-com/am9icG9zdDr5B-u5pDsIyaoOntQlyyXz
++++
As Atla’s Founding Machine Learning Engineer, you’ll spearhead the scaling of our fine-tuning and inference frameworks for large language models.
Qualifications
Evidence of exceptional ML engineering ability:
You are going to thrive at Atla with the following mindset:
Please apply to this role on our ATS
https://jobs.gem.com/atla-ai-com/am9icG9zdDr5B-u5pDsIyaoOntQlyyXz
++++
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.
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.
- 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.
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.
- Competitive salary
- Significant stake in equity as one of the first joiners