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

Cohere
Paris, London, Toronto, New York, Montreal
Workplace: RemoteFull timeFunction: Research & Scientific (R&D)Skills: ["Collaboration","Teamwork","Communication","Problem-solving"]

Design and implement high-performing, scalable software for training AI models; build tools to accelerate research and LLM training; collaborate across Infrastructure, Efficiency, Serving, and scientific teams to create an integrated post-training ecosystem; optimize training cycles for SFT, offline preference, and RL; explore cluster and data infrastructure ideas to boost scalability and performance.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Cohere
Cohere
1 year ago

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

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 18 hours agoStatus: Live

Job Summary

Design and implement high-performing, scalable software for training AI models; build tools to accelerate research and LLM training; collaborate across Infrastructure, Efficiency, Serving, and scientific teams to create an integrated post-training ecosystem; optimize training cycles for SFT, offline preference, and RL; explore cluster and data infrastructure ideas to boost scalability and performance.
Location: Paris, London, Toronto, New York, Montreal
Workplace: Remote
Employment Type: Full time
Job Function: Research & Scientific (R&D)

Key Responsibilities

  • •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 (Infrastructure, Efficiency, Serving) and the scientific teams (Agent, Multimodal, Multilingual, etc.) to create a strong and integrated post-training ecosystem.
  • •Craft and implement techniques to improve performance and speed up our training cycles, both on SFT, offline preference, and the RL regime.
  • •Research, implement, and experiment with ideas on our cluster and data infrastructure.

Pay and Benefits

Perks:Health InsuranceDentalParental LeavePaid LeaveRemote Work

Key Requirements

  • •Extremely strong software engineering skills.
  • •Proficiency in Python and ML frameworks such as JAX, PyTorch and/or XLA/MLIR.
  • •Experience using and debugging large-scale distributed training strategies (memory/speed profiling).
  • •[Bonus] Experience with distributed training infrastructures (Kubernetes) and associated frameworks (Ray).
  • •[Bonus] Hands-on experience with the post-training phase of model training, with a strong emphasis on scalability and performance
Experience:AI/MLLLMRL
Skills:CollaborationTeamworkCommunicationProblem-solving
Tech Stack:PythonJAXPyTorchXLA/MLIRKubernetesRay

Company Brief

Cohere
Builds security-first foundation models and enterprise AI products (LLMs, retrieval, agent platforms) for regulated industries, enabling customizable, private deployments across cloud and on-premises for real-world business applications.
Industry: AI & Machine Learning
Company Size: Large (251 to 1,000 employees)
Growth: Scaleup
Valuation: Unicorn (USD 1B+)
Funding: Series D
Headquarters: Toronto, Canada
Founded: 2019
Glassdoor
Glassdoor: 2.9
WebsiteLinkedInGlassdoor