Member of Technical Staff, Modeling

Cohere Cohere · AI Frontier · London, United Kingdom · Modeling

Cohere is seeking a Member of Technical Staff, Modeling to design, build, and scale AI systems for serving users. This role involves researching, implementing, and experimenting with ideas on supercompute and data infrastructure, with a strong emphasis on both research and production code. The position requires strong software engineering skills, proficiency in Python and ML frameworks, experience with large-scale distributed training, and GPU programming.

What you'd actually do

  1. Design, build and scale AI systems for serving our users.
  2. Research, implement, and experiment with ideas on our supercompute and data infrastructure.
  3. Learn from and work with the best researchers in the field.

Skills

Required

  • Python
  • Tensorflow
  • TF-Serving
  • JAX
  • XLA/MLIR
  • CUDA
  • large-scale distributed training
  • autoregressive sequence models
  • Transformers

Nice to have

  • paper at top-tier venues (such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, EMNLP)

What the JD emphasized

  • Extremely strong software engineering skills.

Other signals

  • training and deploying frontier models
  • scale intelligence
  • supercompute and data infrastructure
  • production code
  • research