Member of Technical Staff, Modeling - AI/ML
CohereWho are we?
Cohere is on a mission to scale intelligence to serve humanity. We are dedicated to training and deploying frontier models for developers and enterprises, enabling them to build AI systems for content generation, semantic search, and more. Our commitment to innovation drives us to obsess over the quality and impact of our work.
Why this role?
As a Member of Technical Staff, you will be at the forefront of AI and ML, designing and implementing novel research ideas and shipping state-of-the-art models to production. You will have access to one of the highest ratios of compute to engineers in the world, working in an environment that blurs the lines between engineering and research.
Responsibilities
- Design, build, and scale AI systems for serving our users.
- Research, implement, and experiment with ideas on our supercompute and data infrastructure.
- Learn from and work with the best researchers in the field.
Who you are
- 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 training on TPUs.
- Experience using large-scale distributed training strategies.
- Familiarity with autoregressive sequence models, such as Transformers.
Perks
- An open and inclusive culture and work environment.
- Work closely with a team on the cutting edge of AI research.
- Comprehensive health and dental benefits, including a separate budget to take care of your mental health.
- 100% Parental Leave top-up for 6 months for employees based in Canada, the US, and the UK.
- Personal enrichment benefits towards arts and culture, fitness and well-being, quality time, and workspace improvement.
- Remote-flexible with offices in major cities and a co-working stipend.
- Generous vacation policy with 6 weeks of vacation.
Location
This role is hybrid, based in San Francisco, CA, with flexibility regarding your location.
Benefits Extracted with AI
- Weekly lunch stipend
- In-office lunches & snacks
- Full health and dental benefits
- Mental health budget
- Parental Leave top-up for 6 months
- Personal enrichment benefits
- Co-working stipend
- 6 weeks of vacation
Similar jobs
Last update: 23 minutes ago
Member of Technical Staff, AI and Machine Learning
Join Cohere as a Member of Technical Staff to design and scale AI systems, focusing on AI, ML, and TensorFlow.
Senior Data Annotator/AI Data Trainer - Coding
Join Cohere as a Senior Data Annotator/AI Data Trainer focusing on coding. Enhance AI models through data quality improvement.
Machine Learning Engineer - Ads
Join as a Machine Learning Engineer focusing on Ads, developing predictive models in a hybrid role in New York.
Machine Learning Engineer with AI/ML Experience
Join us as a Machine Learning Engineer to develop AI/ML models and applications. Work remotely with top-tier companies.
AI Framework Engineer
Join webAI as an AI Framework Engineer to develop innovative AI frameworks for distributed computing environments.
Senior Applied AI Researcher
Join CHAI: AI Platform as a Senior Applied AI Researcher to optimize and innovate AI solutions in a high-growth environment.
Senior ML Infrastructure Engineer
Join CHAI: AI Platform as a Senior ML Infrastructure Engineer to build and scale ML systems in Palo Alto.
Machine Learning Engineer
Join Ema Unlimited as a Machine Learning Engineer in SF Bay Area, working on cutting-edge AI solutions with a focus on NLP and ML technologies.
AI Software Development Engineer
Join Intel as an AI Software Development Engineer to develop and deploy AI applications, enhancing engineering productivity.
Software Engineer - Member of Technical Staff
Join FutureHouse as a Software Engineer to innovate AI systems for scientific research in San Francisco.
Senior Full-stack Engineer
Join Cascading AI as a Senior Full-stack Engineer to develop AI-driven lending solutions in San Francisco.
AI Solutions Software Engineer
Join DwellFi as an AI Solutions Software Engineer to develop innovative AI solutions using LangChain or Llama.
Machine Learning Compiler Engineer
Join Qualcomm as a Machine Learning Compiler Engineer to optimize ML compilers for cutting-edge accelerators.
Research Engineer, Language - Generative AI
Join Meta as a Research Engineer in Generative AI, focusing on large language models and NLP.
Internship Software Engineer - Machine Learning Infrastructure
Join Tesla as an Internship Software Engineer in Machine Learning Infrastructure. Work on AI infrastructure and neural network scaling.
AI/ML/LLM Proof of Concept Engineer
Join Cisco as an AI/ML/LLM Proof of Concept Engineer to develop and demonstrate innovative AI solutions.
Senior Prompt Engineer
Join Accrete AI as a Senior Prompt Engineer to design and optimize prompts for AI agents, enhancing NLP applications.
Applied AI Scientist
Join ClimateAi as an Applied AI Scientist to develop AI solutions for climate resilience. Work with diverse teams in a hybrid environment.
AI Engineer Intern, Self-Driving
Join Tesla as an AI Engineer Intern to develop large-scale models for self-driving technology. Work on cutting-edge AI techniques.
Founding Applied AI Engineer
Join LlamaIndex as a Founding Applied AI Engineer to build and deploy LLM applications. Competitive salary and equity offered.
AI Engineer Intern - Export & Inference
Join Tesla as an AI Engineer Intern focusing on Export & Inference. Work on cutting-edge AI projects in Palo Alto.
Technical Program Manager, External Technical Partnerships
Join Anthropic as a Technical Program Manager to drive cloud deployments and partnerships with Amazon Bedrock and GCP Vertex AI.
Mid-Level AI Researcher
Join Leonardo.Ai as a Mid-Level AI Researcher to develop and refine AI models, focusing on model training and optimization.
Senior Distributed Systems Engineer
Join webAI as a Senior Distributed Systems Engineer to design and maintain scalable systems using Python, Kubernetes, and more.