Member of Technical Staff, Modeling
Member of Technical Staff, Modeling
Cohere
Cohere is a forward-thinking company focused on scaling intelligence to enhance AI systems for developers and enterprises. They are seeking a Member of Technical Staff to design and implement AI models, contributing to both research and production code. The role is remote-friendly and emphasizes collaboration with top researchers in the field.
Qualification
- Extremely strong software engineering skills.
- Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR.
- Experience writing kernels for GPUs using CUDA.
- Experience using large-scale distributed training strategies.
- Familiarity with autoregressive sequence models, such as Transformers.
Responsibility
- Design, build and scale AI systems for serving users.
- Research, implement, and experiment with ideas on supercompute and data infrastructure.
- Contribute to writing production code and conducting research based on individual interests and organizational needs.
- Collaborate with leading researchers in the AI field.
- Maintain deep connections to academia.
