Member of Technical Staff, Post-Training

Cohere
London, Paris, Toronto, New York, Montreal, San Francisco
Workplace: RemoteFull timeFunction: Education & TrainingSkills: ["Collaboration","Communication","Problem-solving"]

Join Cohere as a Member of Technical Staff focusing on post-training software for large-scale model training. You’ll write high-performance, scalable code, push models to SOTA levels, coordinate with cross-functional teams, optimize training cycles in SFT/RL modes, and work with a world-class research team on cutting-edge compute and data infrastructure.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Cohere
Cohere
1 year ago

Member of Technical Staff, Post-Training

✓ Verified Job

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

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

Job Summary

Join Cohere as a Member of Technical Staff focusing on post-training software for large-scale model training. You’ll write high-performance, scalable code, push models to SOTA levels, coordinate with cross-functional teams, optimize training cycles in SFT/RL modes, and work with a world-class research team on cutting-edge compute and data infrastructure.
Location: London, Paris, Toronto, New York, Montreal, San Francisco
Workplace: Remote
Employment Type: Full time
Job Function: Education & Training

Key Responsibilities

  • •Design and write high-performant and scalable software for training models.
  • •Consistently post-train the models to reach SOTA level performance.
  • •Coordinate with other specialist teams (Agentic, Code…) to produce models that have strong all encompassing performance.
  • •Craft and implement techniques to improve the performance and results of our training cycles both on the SFT and the RL regime.
  • •Research, implement, and experiment with ideas on our supercompute and data infrastructure.

Pay and Benefits

Perks:Health InsuranceDentalRemote WorkParental LeaveMeal Allowance

Key Requirements

  • •Extremely strong software engineering skills.
  • •Proficiency in Python and related ML frameworks such as JAX, PyTorch and XLA/MLIR.
  • •Experience with distributed training infrastructures (Kubernetes, Slurm) and associated frameworks (Ray).
  • •Experience using large-scale distributed training strategies.
  • •Hands on experience on training large model at scale and post-training phase with a strong emphasis on performance optimisation.
Experience:AIMachine LearningDistributed trainingResearch & Development
Skills:CollaborationCommunicationProblem-solving
Tech Stack:PythonJAXPyTorchXLA/MLIRKubernetesSlurmRay

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