Member of Technical Staff, Agentic Environments

This is a fully remote position, open to applicants in Europe.

📋 Description

• Create and develop high-performing, scalable software for training models.

• Build tools that enhance and expedite research and LLM training processes.

• Collaborate with teams in Infrastructure, Efficiency, Serving, and scientific areas including Agent, Multimodal, and Multilingual.

• Investigate, implement, and test innovative concepts related to cluster and data infrastructure.

• Partner with scientists, engineers, and other teams to drive projects forward.

• Engage with production systems and experiment with cutting-edge models.

• Design experiments and create documentation for scalable, production-ready solutions.

⛳️ Requirements

• Exceptional software engineering abilities.

• Commitment to test-driven development, clean coding practices, and minimizing technical debt.

• Expertise in Python programming.

• Proficient with ML frameworks such as JAX, PyTorch, and/or XLA/MLIR.

• Familiarity with large-scale distributed training strategies, including memory and speed profiling.

• Experience with distributed training infrastructure like Kubernetes.

• Knowledge of related frameworks such as Ray.

• Background in ML and LLM academic research.

• A strong passion for delivering high-quality work.

• Comfortable with tuning and optimizing large LLM models.

• Able to collaborate with individuals of varying software engineering skill levels.

• Skilled at navigating complex ML codebases to identify and resolve challenges.

• Capable of thriving in a fast-paced and technically demanding environment.

🏝️ Benefits

• A weekly lunch stipend of $75/£75 or its equivalent in your local currency.

• Comprehensive health and dental benefits, alongside a dedicated budget for mental health.

• RRSP matching, 401K, and a Pension Scheme.

• 100% parental leave top-up for up to 6 months, applicable to either parent.

• Annual enrichment benefits for arts and culture, fitness/wellness, quality time, and workspace enhancements.

• Education and learning stipend for conferences, courses, and coaching opportunities.

• Six weeks of paid vacation (30 working days).

• Budget for travel to other offices for remote employees.

• Annual company offsite event.

• Daily lunch program, snacks, and frequent community and social events for employees in the office.

• Co-working benefit for employees located away from an office.

• $500 home office stipend.