Cohere20.02.2025
Member of Technical Staff, Training Infra Engineer
Зарплата не указана
Полная занятостьУдалёнка
Обязанности
- 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
Требования
- 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)
Условия
- 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