Cohere

Member of Technical Staff, Modeling

Cohere
Apply
almost 2 years ago
Toronto, Canada +5 moreSenior
H1B Sponsor

Responsibilities

  • Design, build, and scale AI systems for users.
  • Research, implement, and experiment with ideas using supercompute and data infrastructure.
  • Ship state-of-the-art models to production.
  • Contribute to both production coding and research according to individual interests and organizational needs.
  • Work with leading researchers in the field and maintain connections to academia.

Requirements

  • Extremely strong software engineering skills.
  • Proficiency in Python and machine-learning frameworks including TensorFlow, TF-Serving, JAX, and XLA/MLIR.
  • Experience writing GPU kernels using CUDA.
  • Experience with large-scale distributed training strategies.
  • Familiarity with autoregressive sequence models such as Transformers.
  • A publication at a top-tier venue such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, or EMNLP is a bonus.
  • The listed attributes are neither exhaustive nor required, and candidates with varied backgrounds are encouraged to apply.

Benefits

  • Weekly lunch stipend, in-office lunches, and snacks.
  • Full health and dental benefits, including a separate mental-health budget.
  • 100% parental-leave top-up for up to 6 months.
  • Personal enrichment benefits for arts and culture, fitness and well-being, quality time, and workspace improvement.
  • Remote-flexible work with offices in Toronto, New York, San Francisco, London, and Paris, plus a co-working stipend.
  • Six weeks of vacation, or 30 working days.

Tech Stack

PythonTensorFlow

Categories

AI ResearchML Engineering
Cohere

About Cohere

201-500 employees

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