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

Hiring organization: [Cohere](https://career.thegoodapps.co/organizations/cohere)

Canonical page: https://career.thegoodapps.co/jobs/81d02eea-effa-4fce-8764-17e994b179ed

Listed on Cohere's own careers site. Applications go to them directly.

- Employment type: full time
- Location: New York, NY
- Remote: yes
- Salary: 250000 – 535000 CAD per year

## Summary

This role focuses on developing and scaling machine learning infrastructure for large-language model training, particularly reinforcement learning methods. It suits experienced software engineers who combine strong ML fundamentals with distributed systems expertise and want to contribute to cutting-edge AI infrastructure at an enterprise scale.

_Our summary, not Cohere's wording._

## Skills named

JAX, Kubernetes, Python, PyTorch, Reinforcement Learning, V-Ray

## Required

- Strong software engineering fundamentals
- Proficiency in Python and ML frameworks (JAX, PyTorch, or XLA/MLIR)
- Experience debugging and profiling large-scale distributed training
- Test-driven development and clean code practices

## Nice to have

- Experience with distributed training infrastructures like Kubernetes or Ray
- Hands-on post-training experience with emphasis on scalability and performance
- Background in ML, LLM, or RL academic research

Apply on Cohere's site: https://jobs.ashbyhq.com/cohere/e9ef9420-88e5-403b-82e8-1c36fa010d4d/application