Member of Technical Staff, Agentic Environments

📋 Description

• 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 and scientific teams.
• Research, implement, and experiment with ideas on our cluster and data infrastructure.
• Collaborate with scientists, engineers, and teams to drive production-ready AI systems.

🎯 Requirements

• Extremely strong software engineering skills.
• Value test-driven development, clean code, and reduce technical debt.
• Proficiency in Python and ML frameworks like JAX, PyTorch and/or XLA/MLIR.
• Experience with large-scale distributed training strategies (memory/speed profiling).
• Experience with distributed training infrastructures (Kubernetes) and Ray.
• Experience in ML and LLM academic research.

🎁 Benefits

• Perks as described include lunch stipend, health benefits, RRSP/401K, parental leave, enrichment
• Remote-friendly with global offices and home office stipend.