Member of Technical Staff Agentic Environments
Toronto-founded enterprise AI company building foundation models, agentic AI, enterprise search, and private deployment products.
About Cohere
Cohere provides enterprise-focused AI models and end-to-end products, including North for agentic workplace AI, Compass for enterprise search, and models for generation and retrieval. It supports private, VPC, and on-premises deployment options.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will design and build scalable software for training models and create tools that accelerate research and language-model training. You will experiment on cluster and data infrastructure, coordinate with engineering and scientific teams, and help resolve issues in complex machine-learning codebases.
Requirements
- Software engineering
- Test-driven development
- Clean code
- Technical debt reduction
- Python
- JAX
- PyTorch
- XLA
- MLIR
- Distributed training
- Memory profiling
- Speed profiling
- Kubernetes
- Ray
- Machine learning research
- Language model research
Responsibilities
- Design and write scalable model-training software
- Develop tools that accelerate research and language-model training
- Coordinate with engineering and scientific teams
- Research, implement, and experiment on cluster and data infrastructure
- Collaborate with scientists, engineers, and teams
Benefits
- Weekly lunch stipend of $75/£75 or local-currency equivalent
- Health and dental benefits
- Mental health budget
- RRSP matching
- 401K
- Pension Scheme
- Parental leave top-up for up to six months
- Arts and culture benefit
- Fitness and wellness benefit
- Workspace improvement credit
- Education and learning stipend
- Six weeks of paid vacation
- Remote travel budget to other offices
- Annual company offsite
- Daily lunch program for office-based employees
- Snacks for office-based employees
- Community and social events for office-based employees
- Co-working benefit
- $500 home office stipend