C
Member of Technical Staff, Integration/RL Team (Research Engineer)
CohereNew York, NY · Remote · full time
CA$250,000 – CA$535,000
Listed on Cohere’s own careers site. You apply with them directly — we never stand between you and the employer.
What this role is
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. The full posting is on their site.
Skills this role names
Log in to see which of these are already on your profile.
What they ask for
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