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!)
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!
