# Member of Technical Staff, Integration/RL Team

**Company:** [Cohere](https://hotfix.jobs/companies/cohere)
**Location:** Remote
**Role:** ML Engineering
**Skills:** Python, JAX, PyTorch, Xla, Mlir, Kubernetes, Ray, Distributed Training, Reinforcement Learning, LLMs, Machine Learning, Test-Driven Development, Memory Profiling, Speed Profiling, Cluster Infrastructure
**Posted:** 2025-08-19

> Build and scale high-performance software and distributed reinforcement-learning systems for LLM post-training. The role combines production engineering, research tooling, algorithm optimization, and collaboration across infrastructure and scientific teams.

## Job Description

## Responsibilities
- Design and write high-performing, scalable software for training models.
- Develop tools to support and accelerate research and LLM training.
- Coordinate with infrastructure, efficiency, serving, and scientific teams to build an integrated post-training ecosystem.
- Implement techniques to improve performance and accelerate training cycles across supervised fine-tuning (SFT), offline preference optimization, and reinforcement learning.
- Research, implement, and experiment with ideas on cluster and data infrastructure.
- Collaborate with scientists, engineers, and cross-functional teams.

## Requirements
- Extremely strong software engineering skills.
- Value test-driven development, clean code, and reducing technical debt.
- Proficiency in Python and related machine-learning frameworks such as JAX, PyTorch, and/or XLA/MLIR.
- Experience using and debugging large-scale distributed training strategies, including memory and speed profiling.

## Nice-to-haves
- Experience with distributed training infrastructure such as Kubernetes and associated frameworks such as Ray.
- Hands-on experience with model post-training, particularly scalability and performance.
- Experience in machine-learning, LLM, and reinforcement-learning academic research.

## Benefits
- Weekly lunch stipend of $75/£75 or equivalent in local currency.
- Full health and dental benefits, including a separate mental-health budget.
- RRSP matching, 401(k), or pension scheme.
- Up to six months of 100% parental-leave top-up for either parent.
- Annual enrichment benefits for arts and culture, fitness and wellness, quality time, and workspace improvements.
- Education and learning stipend for conferences, courses, and coaching.
- Six weeks of paid vacation.
- Travel budget for remote employees and an annual company offsite.
- Coworking benefit for employees not near an office.
- $500 home-office stipend.

## Similar jobs

- [Research Software Engineer, Post Training](https://hotfix.jobs/jobs/168e3c8f-8577-4482-bf94-91b3d11744ba) - Thinking Machines Lab - San Francisco, CA - $350k – $475k/yr
- [Software Engineer, AI for Chip Design](https://hotfix.jobs/jobs/abfb017d-1ce2-4d24-901d-16084eb7b3bc) - OpenAI - San Francisco, CA - $266k – $468k/yr
- [AI Software Engineer](https://hotfix.jobs/jobs/c9a0e889-a36e-488f-87d0-e9c27ab63bf0) - Rollstack - Remote
- [Machine Learning Engineer, Ranking & Retrieval](https://hotfix.jobs/jobs/adb5411d-8bfc-48b8-a8e7-15cbf5ea21b2) - ClickUp - Remote - $200k – $250k/yr
- [Machine Learning Engineer III](https://hotfix.jobs/jobs/2359be26-5005-4fa7-94c9-8a86066a6bb5) - PathAI - Boston, MA - $131k – $200k/yr

**Apply:** https://hotfix.jobs/jobs/8bfad1e4-4a7e-4d26-a297-19a34c40520c
**Canonical:** https://hotfix.jobs/jobs/8bfad1e4-4a7e-4d26-a297-19a34c40520c