Member of Technical Staff, Agentic Environments
Remote • Europe • FullTime
Posted 1d ago
Job Location
Europe
Tech Stack
Remote Work Policy
Fully remote
Employment Type
FullTime
Categories
LLM Engineer
About the job
Cohere is seeking a senior engineer to join our team, focusing on the practical challenges of deploying AI systems at scale in production environments. This hands-on, engineering-driven role involves working with frontier AI models, building scalable solutions, and bridging research concepts with real-world implementations. You will contribute to both engineering and research efforts, designing and writing high-performing software for model training, developing new tools to support LLM research, and collaborating with various engineering and scientific teams. We provide access to world-class compute resources, data, and talent to enable you to do your best work.
Responsibilities
- Design and write high-performing, scalable software for model training.
- Develop new tools to support and accelerate research and LLM training.
- Coordinate with infrastructure, efficiency, and serving engineering teams, as well as scientific teams.
- Research, implement, and experiment with ideas on cluster and data infrastructure.
- Collaborate with scientists, engineers, and other teams.
Requirements
- Extremely strong software engineering skills.
- Proficiency in test-driven development, clean code, and reducing technical debt.
- Proficiency in Python and ML frameworks like JAX, PyTorch, and/or XLA/MLIR.
- Experience with large-scale distributed training strategies (memory/speed profiling).
- Experience with distributed training infrastructures (Kubernetes) and frameworks (Ray).
- Experience in ML and LLM academic research.
- Passion for quality work.
- Enjoy tuning and optimizing large LLM models.
- Comfortable working with individuals of varying software engineering skill levels.
- Comfortable diving into complex ML codebases to identify and resolve issues.
- Ability to thrive in a fast-paced, technically challenging environment.
Benefits
- Weekly lunch stipend of $75/£75 or local currency equivalent.
- Full health and dental benefits, including a mental health budget.
- RRSP matching, 401K, Pension Scheme.
- 100% Parental Leave top-up for up to 6 months.
- Annual enrichment benefits (Arts & culture, fitness/wellness, quality time, workspace improvement).
- Education & learning stipend for conferences, courses, and coaching.
- 6 weeks (30 working days) of paid vacation.
- Budget for traveling to other offices (if remote).
- Annual company offsite.
- $500 home office stipend.