Staff Engineer, AI Training Infra

Cohere; Toronto; Remoto; CAD 250.000 - 535.000; Tempo pieno

Vantaggi offerti da questo lavoro

Lunch stipend

Health benefits

RRSP matching

Parental leave

Enrichment benefits

Education stipend

Vacation and travel budget

Home office stipend

Descrizione del lavoro

Cohere is hiring a Senior ML Engineer to design scalable training systems, ship state-of-the-art models to production, and bridge research with production code. You will work across infrastructure, tooling, and data platforms to accelerate training at scale.

We offer a remote-friendly setup with offices worldwide, competitive compensation, health benefits, retirement plans, and continual learning support to help you grow.

Competenze

• Extremely strong software engineering skills.
• Proficiency in Python and ML frameworks such as JAX, PyTorch and XLA/MLIR.
• Experience with distributed training infrastructures (Kubernetes, Slurm) and associated frameworks (Ray).
• Experience using large-scale distributed training strategies.
• Hands on experience on training large model at scale and having contributed to the tooling and/or setup of the training infrastructure.
• Bonus: paper at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP).

Mansioni

• Design and write high-performant and scalable software for training.
• Improve our training setup from an infrastructure and codebase performance standpoint.
• Craft and implement tools to speed up our training cycles and improve the overall efficacy of our training infrastructure
• Research, implement, and experiment with ideas on our supercompute and data infrastructure.
• Learn from and work with the best researchers in the field.

Conoscenze

Software engineering

Python

ML frameworks

Distributed training

Large-scale training

Formazione

Bachelor's degree in CS/ML or related

Strumenti

Kubernetes

Slurm

Ray

Descrizione del lavoro

Cohere is hiring a Senior ML Engineer to design scalable training systems, ship state-of-the-art models to production, and bridge research with production code. You will work across infrastructure, tooling, and data platforms to accelerate training at scale.

We offer a remote-friendly setup with offices worldwide, competitive compensation, health benefits, retirement plans, and continual learning support to help you grow.