Research Software Engineer - ML Training Infrastructure

Reflection; Greater London; Sur place; GBP 70 000 - 100 000; Plein temps

Avantages offerts par ce poste

Top-tier compensation

Comprehensive medical, dental, vision insurance

Fully paid parental leave

Paid time off when needed

Daily lunch and dinner provided

Résumé du poste

Reflection, based in the United Kingdom, is seeking a Software Engineer to architect and optimize the training infrastructure for AI models. The role focuses on building scalable systems for reinforcement learning and distributed training, requiring deep experience in distributed systems. Candidates should have practical skills in tools like PyTorch and JAX, and a robust understanding of performance optimization. The position offers top-tier compensation and comprehensive health benefits, alongside a supportive work environment.

Qualifications

• Strong software engineering skills in machine learning.
• Experience in distributed training or data infrastructure.
• Ability to implement research papers in practical applications.

Responsabilités

• Design and optimize large-scale training loops and data pipelines.
• Implement state-of-the-art techniques ensuring stability and efficiency.
• Build internal tooling for monitoring and reproducing experiments.

Connaissances

Distributed Training & Inference

Data Infrastructure

Performance optimization

Numerical stability

Machine learning

Outils

PyTorch

JAX

Ray

Kubernetes

Slurm

Description du poste

Reflection, based in the United Kingdom, is seeking a Software Engineer to architect and optimize the training infrastructure for AI models. The role focuses on building scalable systems for reinforcement learning and distributed training, requiring deep experience in distributed systems. Candidates should have practical skills in tools like PyTorch and JAX, and a robust understanding of performance optimization. The position offers top-tier compensation and comprehensive health benefits, alongside a supportive work environment.