[Remote] Member of Technical Staff, Modeling

Note: The job is a remote job and is open to candidates in USA. Cohere is a company focused on scaling intelligence to serve humanity through advanced AI systems. They are seeking a Member of Technical Staff to design and implement innovative research ideas, develop state-of-the-art models, and collaborate closely with researchers.ResponsibilitiesDesign, build and scale AI systems for serving our usersResearch, implement, and experiment with ideas on our supercompute and data infrastructureLearn from and work with the best researchers in the fieldSkillsExtremely strong software engineering skillsProficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIRExperience writing kernels for GPUs using CUDAExperience using large-scale distributed training strategiesFamiliarity with autoregressive sequence models, such as TransformersPaper at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP)BenefitsAn open and inclusive culture and work environmentWork closely with a team on the cutting edge of AI researchWeekly lunch stipend, in-office lunches & snacksFull health and dental benefits, including a separate budget to take care of your mental health100% Parental Leave top-up for up to 6 monthsPersonal enrichment benefits towards arts and culture, fitness and well-being, quality time, and workspace improvementRemote-flexible, offices in Toronto, New York, San Francisco, London and Paris, as well as a co-working stipend6 weeks of vacation (30 working days!)Company OverviewCohere develops enterprise artificial intelligence software and provides language models, retrieval tools, and workplace platforms. It was founded in 2019, and is headquartered in Toronto, Ontario, CAN, with a workforce of 201-500 employees. Its website is https://cohere.com.Company H1B SponsorshipCohere has a track record of offering H1B sponsorships, with 11 in 2025, 14 in 2024, 13 in 2023, 5 in 2022, 2 in 2021. Please note that this does not guarantee sponsorship for this specific role.