Hiring: Member of Technical Staff, Training Performance Engineer Company: Cohere Location: New York, NY Job Posted Time: 2026-09-02 18:05:35 Employment Type: Full-time / Remote Target Skills & Keywords : Machine Learning, NLP, PyTorch, Python, Transformers, Triton About the job Required Skills: •Design and write high-performant and scalable software for training. •Understand architectural modifications and design choices and their effects on training throughput and quality. •Write low-level CUDA, triton kernels to squeeze every last bit of performance from our accelerators. •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. •Experience writing kernels for GPUs using CUDA, triton, etc •Experience using large-scale distributed training strategies. •Familiarity with autoregressive sequence models, such as Transformers. •: paper at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP). •Working location: •This role can be based remotely or from one of our office locations listed on the job description - there is no minimum in-office qualification requirement. We care most about hiring exceptional people regardless of locations, though please check the location listed on the posting for guidance around the core time zone or working hours alignment expected for the role. •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: •$75 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!