Cohere02/20/2025

Инженер по инфраструктуре для обучения моделей

Salary not specified
MARKET
15,900median for this role
Data Scientist · 115 jobs with disclosed pay
5,000half of the offers: 12,796–20,50043,793
The employer didn't disclose pay — compare with the market yourself.
Полная занятостьУдалёнка

Responsibilities

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

Requirements

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

What we offer

  • 01Weekly lunch stipend of $75/£75 or equivalent in local currency
  • 02Full health and dental benefits, including mental health budget
  • 03RRSP matching, 401K, Pension Scheme
  • 04100% Parental Leave top-up for up to 6 months
  • 05Annual enrichment benefits for arts & culture, fitness, quality time, and workspace improvement
  • 06Education & learning stipend for conferences, courses, and coaching
  • 076 weeks of paid vacation (30 working days)
  • 08Budget for traveling to other offices if remote, plus annual company offsite
  • 09Remote-friendly with offices in multiple cities
  • 10Co-working benefit for those not near an office
  • 11$500 home office stipend