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.

Apply on employer's website

This employer gathers applications via their own applicant tracking system.

You will be redirected to an external application form.

Share job

Meet JobCopilot: Your Personal AI Job Hunter

Automatically Apply to Engineering Jobs. Just set your preferences and Job Copilot will do the rest — finding, filtering, and applying while you focus on what matters.

Activate JobCopilot