Senior/Full-time/EU Remote/United Kingdom (+ 3 more)
Member of Technical Staff, Training Performance Engineer
CohereEU RemoteQuick apply
Role Opportunity Score
57Good opportunity
Niche stackLonger application
Expected applicants: ~30–50
Python and JAX expertise in model training performance optimization sets this role apart, attracting around 30-50 qualified candidates globally. The demand for proficiency in MLIR and XLA makes this field even narrower, as many applicants lack this specialized knowledge.
Evidence Dossier
Remote Details
Scope: EU Remote
Role Summary
As a Performance Engineer in the Pre-Training team, responsible for optimizing performance of advanced language models and systems, improving training throughput, ensuring high accelerator utilization, designing scalable software for training, and working with the best researchers in natural language processing.
Stack Required
PythonJAXPyTorchXLAMLIRCUDATriton
Compensation
Salary not disclosed
weekly lunch stipendfull health and dental benefits including mental health budgetRRSP matching+7 more
Requirements
Languages: English
Visa sponsorship not stated
Equity Offered
AI Tools Culture
AI mentionedLast seen on career page: 28d ago
Posted: 1mo ago