🛠 Tech Stack

💼 About This Role

You'll build and optimise large-scale learning systems for open-weight models at a pioneering AI company. You'll work hand-in-hand with Research Scientists on distributed training or data pipelines, and turn fresh ideas into repeatable, scalable code. This role offers the flexibility to move along the research-to-production spectrum as interests evolve.

🎯 What You'll Do

• Accelerate researchers by building robust ML pipeline tools.
• Integrate checkpoints, streamline evaluation, and expose APIs.
• Conduct experiments with distributed training on thousands of GPUs.
• Design and benchmark ML algorithms in Python.

📋 Requirements

• Master's or PhD in Computer Science or equivalent.
• 4+ years working on large-scale ML codebases.
• Hands-on with PyTorch, JAX, or TensorFlow.
• Experience with distributed training (DeepSpeed, FSDP, SLURM, K8s).

✨ Nice to Have

• Experience in deep learning, NLP, or LLMs.
• CUDA or data-pipeline skills.
• Strong software-design instincts (testing, code review, CI/CD).

🎁 Benefits & Perks

• 💰 Competitive cash salary and equity
• 🥕 Daily lunch vouchers
• 🥎 Monthly contribution to Gympass subscription
• 🚴 Monthly contribution to mobility pass
• 🧑‍⚕️ Full health insurance for you and family