Hiring: Member of Technical Staff, Training Infra Engineer Company: Cohere Location: San Francisco, CA Job Posted Time: 2026-09-10 10:17:39 Employment Type: Full-time / Remote Target Skills & Keywords : CV, Kubernetes, PyTorch, Python About the job Required Skills: •Please Note: We have offices in London, Paris, Toronto, San Francisco and New York but also embrace being remote-friendly! There are no restrictions on where you can be located for this role. •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. Qualifications: •Extremely strong software engineering skills. •Proficiency in Python and related ML frameworks such as JAX, Pytorch and XLA/MLIR. •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). •Full-Time Employees At Cohere Enjoy These Perks •A weekly lunch stipend of $75/£75 or equivalent in your local currency for lunch. •Full health and dental benefits, including a separate budget for mental health. •RRSP matching, 401K, Pension Scheme. •100% Parental Leave top-up for up to 6 months, for either parent. Compensation: •Flexible work environment (work from home / hybrid options) Interested candidates, please apply directly through the job posting on company's career page or try via AI auto apply on this platform. Don't miss this opportunity to join a forward-thinking team!