Taro Logo

Member of Technical Staff, Integration/RL Team (Research Engineer)

AI company training and deploying frontier models for developers and enterprises to power content generation, semantic search, RAG, and agents.
Machine Learning
Senior Software Engineer
Remote
AI

Job Description

Cohere is at the forefront of AI development, training and deploying frontier models for enterprise applications. The Integration team seeks a Member of Technical Staff to focus on LLM post-training and distributed RL methods. This role combines engineering excellence with scientific rigor, requiring expertise in developing scalable ML software and optimizing training algorithms. The position offers the opportunity to work with cutting-edge AI technology while collaborating with world-class researchers and engineers. The ideal candidate will have strong software engineering skills, experience with ML frameworks, and a passion for optimization. The role supports Cohere's mission to scale intelligence to serve humanity, offering the chance to work on unprecedented challenges in AI development. The position provides comprehensive benefits, flexible work arrangements, and the opportunity to shape the future of AI technology. Located in major tech hubs or remote-friendly between UTC−06:00 and UTC+01:00, this role offers both technical challenges and the chance to contribute to groundbreaking AI advancements.

Last updated 8 hours ago

Responsibilities For Member of Technical Staff, Integration/RL Team (Research Engineer)

  • Design and write high-performing and scalable software for training models
  • Develop new tools to support and accelerate research and LLM training
  • Coordinate with other engineering teams to create a strong integrated post-training ecosystem
  • Implement techniques to improve performance and speed up training cycles
  • Research, implement, and experiment with ideas on cluster and data infrastructure
  • Collaborate with scientists, engineers, and other teams

Requirements For Member of Technical Staff, Integration/RL Team (Research Engineer)

Python
  • Extremely strong software engineering skills
  • Value test-driven development methods and clean code
  • Proficiency in Python and ML frameworks (JAX, Pytorch, XLA/MLIR)
  • Experience using and debugging large-scale distributed training strategies
  • Comfortable working with people with different levels of software engineering skills
  • Comfortable diving into complex ML codebases

Benefits For Member of Technical Staff, Integration/RL Team (Research Engineer)

Dental Insurance
Medical Insurance
Mental Health Assistance
Parental Leave
  • Weekly lunch stipend, in-office lunches & snacks
  • Full health and dental benefits
  • Mental health budget
  • 100% Parental Leave top-up for 6 months (Canada, US, UK)
  • Personal enrichment benefits
  • Remote-flexible work
  • Co-working stipend
  • 6 weeks of vacation

Related Jobs