Member of Technical Staff, Integration/RL Team (Research Engineer)

Stack

Responsibilities

• ▹Design and write high-performing and scalable software for training models
• ▹Develop new tools to support and accelerate research and LLM training
• ▹Coordinate with other engineering teams (Infrastructure, Efficiency, Serving) and scientific teams (Agent, Multimodal, Multilingual, etc.) to create an integrated post-training ecosystem
• ▹Craft and implement techniques to improve performance and speed up training cycles across SFT, offline preference and the RL regime
• ▹Research, implement and experiment with ideas on the cluster and data infrastructure
• ▹Collaborate extensively with other scientists, engineers and teams

Requirements

• ▹Extremely strong software engineering skills
• ▹Value test-driven development, clean code and reducing technical debt at all levels
• ▹Proficiency in Python and related ML frameworks such as JAX, PyTorch and/or XLA/MLIR
• ▹Experience using and debugging large-scale distributed training strategies (memory/speed profiling)

Nice to have

• ▹Experience with distributed training infrastructures (Kubernetes) and associated frameworks (Ray)
• ▹Hands-on experience with the post-training phase of model training, with emphasis on scalability and performance
• ▹Experience in ML, LLM and RL academic research

Soft skills

What we offer

• ▹Weekly lunch stipend of $75/£75 (or local equivalent)
• ▹Full health and dental benefits, including a separate mental health budget
• ▹RRSP matching, 401K, Pension Scheme
• ▹100% Parental Leave top-up for up to 6 months for either parent
• ▹Annual enrichment benefits and an education & learning stipend
• ▹6 weeks of paid vacation (30 working days)
• ▹Travel budget to other offices for remote staff plus an annual company offsite
• ▹$500 home office stipend; co-working benefit for those not near an office