← Back to list
Job · Principal

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

Cohere
AI / ML Engineer • Principal • Remote • Full-time • 📍 Paris
📄 View original listing (EN)

Member of Technical Staff (Research Engineer) on Cohere's Integration team, developing and scaling ML algorithms and infrastructure for LLM post-training with a focus on large-scale distributed RL. Remote-friendly for applicants between UTC-06:00 and UTC+01:00.

Stack

AI/MLKubernetesPythonPyTorch

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 scientific teams (Agent, Multimodal, Multilingual, etc.) to create an integrated post-training ecosystem
  • ▹Craft and implement techniques to improve performance and speed up training cycles across SFT, offline preference and the RL regime
  • ▹Research, implement and experiment with ideas on the cluster and data infrastructure
  • ▹Collaborate extensively with other scientists, engineers and teams

Requirements

  • ▹Extremely strong software engineering skills
  • ▹Value test-driven development, clean code and reducing technical debt at all levels
  • ▹Proficiency in Python and related ML frameworks such as JAX, PyTorch and/or XLA/MLIR
  • ▹Experience using and debugging large-scale distributed training strategies (memory/speed profiling)

Nice to have

  • ▹Experience with distributed training infrastructures (Kubernetes) and associated frameworks (Ray)
  • ▹Hands-on experience with the post-training phase of model training, with emphasis on scalability and performance
  • ▹Experience in ML, LLM and RL academic research

Soft skills

Passion for quality workCollaboration with people of different skill levelsThriving in a fast-paced, technically challenging environment

What we offer

  • ▹Weekly lunch stipend of $75/£75 (or local equivalent)
  • ▹Full health and dental benefits, including a separate mental health budget
  • ▹RRSP matching, 401K, Pension Scheme
  • ▹100% Parental Leave top-up for up to 6 months for either parent
  • ▹Annual enrichment benefits and an education & learning stipend
  • ▹6 weeks of paid vacation (30 working days)
  • ▹Travel budget to other offices for remote staff plus an annual company offsite
  • ▹$500 home office stipend; co-working benefit for those not near an office
Languages: angol