6 hours ago
Responsibilities
- Design and write high-performing, scalable software for model training.
- Develop tools that support and accelerate research and LLM training.
- Coordinate with infrastructure, efficiency, serving, and scientific engineering teams.
- Research, implement, and experiment with ideas on cluster and data infrastructure.
- Collaborate with scientists, engineers, and cross-functional teams to build production-ready solutions.
Requirements
- Extremely strong software engineering skills.
- Proficiency in Python and related ML frameworks such as JAX, PyTorch, and/or XLA/MLIR.
- Experience using and debugging large-scale distributed training strategies, including memory and speed profiling.
- Experience with distributed training infrastructure such as Kubernetes and associated frameworks such as Ray.
- Experience in ML and LLM academic research.
- Familiarity with test-driven development methods, clean code, and reducing technical debt.
Benefits
- Weekly lunch stipend of $75/£75 or equivalent in local currency.
- Full health and dental benefits with a separate mental health budget.
- RRSP matching, 401K, and pension scheme.
- 100% parental leave top-up for up to six months for either parent.
- Annual arts and culture, fitness and wellness, quality time, workspace improvement, education, learning, conference, course, and coaching benefits.
- Six weeks of paid vacation, or 30 working days.
- Travel budget for remote employees visiting other offices and an annual company offsite.
- Remote-friendly work arrangement with offices in Toronto, London, New York City, San Francisco, Montreal, Paris, Berlin, and Seoul.
- Daily lunch program, snacks, and community and social events for office employees.
- Co-working benefit for employees not near an office.
- $500 home office stipend.
Tech Stack
Categories
AI ResearchML Engineering
About Cohere
Cohere is the leading security-first enterprise AI company. We build cutting-edge foundation models and end-to-end AI products designed to solve real-world business problems. We partner closely with companies to deliver seamless integration, full customization, and easy-to-use solutions for their workforce and customers. Our all-in-one platform offers enterprises the highest levels of data security, privacy and optionality to deploy across all major cloud providers, private cloud environments, or on-premises. HQ: 171 John Street, 2nd Floor, Toronto, ON M5T 1X3