Member of Technical Staff Modeling
1 hour agoSalary: 250K - 535KLondon; Montreal; New York; Paris; San Francisco; TorontoRemoteFull TimeAiJobs by Cohere
CohereVisit Cohere website
Toronto-founded enterprise AI company building foundation models, agentic AI, enterprise search, and private deployment products.
Toronto, Canada
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
About the Role
You will design and implement research ideas, deploy state-of-the-art models to production, and contribute to both production code and research. You will build and scale AI systems, experiment using supercompute and data infrastructure, and work with researchers on model development.
Requirements
- Strong software engineering skills
- Proficiency in Python and ML frameworks including TensorFlow, TF-Serving, JAX, and XLA/MLIR
- Experience writing GPU kernels with CUDA
- Experience with large-scale distributed training strategies
- Familiarity with autoregressive sequence models such as Transformers
Responsibilities
- Design, build, and scale AI systems for users
- Research, implement, and experiment with ideas on supercompute and data infrastructure
- Contribute to production code and research
Benefits
- Weekly lunch stipend of $75/£75 or local-currency equivalent
- Health and dental benefits
- Mental health budget
- RRSP matching, 401K, and pension scheme
- Parental leave top-up for up to 6 months
- Arts, culture, fitness, wellness, quality-time, and workspace-improvement benefits
- Education and learning stipend
- 6 weeks of paid vacation
- Remote travel budget and annual company offsite
- Co-working benefit
- $500 home office stipend