Cohere · 2 weeks ago
Member of Technical Staff, Integration/RL Team (Research Engineer)
New York, NY
Full-time
Hybrid
Mid LevelCohere is a company dedicated to scaling intelligence to serve humanity through advanced AI systems. The Member of Technical Staff will focus on developing and scaling machine learning algorithms and infrastructure, particularly for large-scale distributed reinforcement learning methods, while collaborating with various engineering and scientific teams.
Artificial Intelligence (AI)Foundational AIGenerative AIMachine LearningNatural Language Processing
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 the scientific teams (Agent, Multimodal, Multilingual, etc.) to create a strong and integrated post-training ecosystem
Craft and implement techniques to improve performance and speed up our training cycles, both on SFT, offline preference, and the RL regime
Research, implement, and experiment with ideas on our cluster and data infrastructure
Collaborate, Collaborate, and Collaborate with other scientists, engineers, and teams!
Qualification
Required
Extremely strong software engineering skills
Value test-driven development methods, clean code, and strive to reduce technical debts 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)
Preferred
Experience with distributed training infrastructures (Kubernetes) and associated frameworks (Ray)
Hands-on experience with the post-training phase of model training, with a strong emphasis on scalability and performance
Experience in ML, LLM and RL academic research
Benefits
An open and inclusive culture and work environment
Work closely with a team on the cutting edge of AI research
Weekly lunch stipend, in-office lunches & snacks
Full health and dental benefits, including a separate budget to take care of your mental health
100% Parental Leave top-up for up to 6 months
Personal enrichment benefits towards arts and culture, fitness and well-being, quality time, and workspace improvement
Remote-flexible, offices in Toronto, New York, San Francisco, London and Paris, as well as a co-working stipend
6 weeks of vacation (30 working days!)
Company
Cohere
Cohere is an enterprise AI firm developing secure and private AI technology to address real-world business challenges.
Founded in 2019
201-500 employees
H1B Sponsorship
Cohere has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (11)
2024 (14)
2023 (13)
2022 (5)
2021 (2)
Funding
Current Stage
Late StageTotal Funding
$1.71BKey Investors
Government of CanadaTiger Global ManagementIndex Ventures
2025-09-24Series D· $100M
2025-08-14Series D· $500M
2025-06-17Secondary Market
Leadership Team
Aidan Gomez
cofounder + ceo
Ivan Zhang
Co-Founder
Recent News
Beyond Bylines
2026-01-11
2026-01-06
Crunchbase News
2026-01-06
Company data provided by crunchbase