Staff ML Engineer, Training Infrastructure

Cohere; À distance; EUR 40 000 - 60 000; Plein temps

Résumé du poste

A cutting-edge AI technology company based in Paris seeks a Member of Technical Staff to develop and enhance training software. You will work on scaling infrastructure and contribute to exciting AI research. Ideal candidates should have strong software engineering skills, proficiency in Python, and experience with distributed training environments. The position is remote-friendly and offers a comprehensive benefits package, including health benefits and generous vacation time.

Prestations

Open and inclusive culture

Weekly lunch stipend

Full health and dental benefits

100% Parental Leave top-up

Personal enrichment benefits

6 weeks of vacation

Qualifications

• Extremely strong software engineering skills required.
• Proficiency in Python and related ML frameworks such as JAX, Pytorch.
• Experience with distributed training infrastructures like Kubernetes and Slurm.

Responsabilités

• Design and write software for training.
• Improve training setup infrastructure and performance.
• Craft tools to speed up training cycles.

Connaissances

Strong software engineering skills

Proficiency in Python

Experience with ML frameworks (JAX, Pytorch)

Distributed training infrastructures (Kubernetes, Slurm)

Experience in large-scale distributed training strategies

Hands-on experience in training large models

Description du poste

A cutting-edge AI technology company based in Paris seeks a Member of Technical Staff to develop and enhance training software. You will work on scaling infrastructure and contribute to exciting AI research. Ideal candidates should have strong software engineering skills, proficiency in Python, and experience with distributed training environments. The position is remote-friendly and offers a comprehensive benefits package, including health benefits and generous vacation time.