# Member of Technical Staff, Modeling

Hiring organization: [Cohere](https://career.thegoodapps.co/organizations/cohere)

Canonical page: https://career.thegoodapps.co/jobs/bc703a5d-61fd-4738-9130-3e9935169781

Listed on Cohere's own careers site. Applications go to them directly.

- Employment type: full time
- Location: New York, NY
- Remote: yes
- Salary: 250000 – 535000 CAD per year

## Summary

This role involves designing and deploying frontier AI models for enterprise use, blending research innovation with production engineering. It suits researchers and engineers who want to work on large-scale foundation models with access to significant compute resources and collaborative teams.

_Our summary, not Cohere's wording._

## Skills named

CUDA, JAX, Python, TensorFlow

## Required

- Extremely strong software engineering skills
- Proficiency in Python and ML frameworks (TensorFlow, TF-Serving, JAX, XLA/MLIR)
- Experience writing GPU kernels using CUDA
- Experience with large-scale distributed training strategies
- Familiarity with autoregressive sequence models like Transformers

## Nice to have

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

Apply on Cohere's site: https://jobs.ashbyhq.com/cohere/3136a5a5-06fd-4c82-8b72-a43467e6b128/application