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

1,001-5,000 employees

Cohere builds large language models and an enterprise AI platform that companies use for search, summarization, and workflow automation, delivered via API or private deployments. Founded in 2019 and headquartered in Toronto, it focuses on multilingual models, data controls, and options to run across major clouds or on-premises. The business is privately held and serves security- and compliance-sensitive organizations.

Contact me