# Member of Technical Staff, Modeling

**Company:** [Cohere](https://hotfix.jobs/companies/cohere)
**Location:** Remote
**Role:** ML Engineering
**Skills:** Python, TensorFlow, tf-serving, JAX, xla, mlir, CUDA, gpu kernels, Distributed Training, Transformers, autoregressive models, supercomputing, Machine Learning
**Posted:** 2024-11-01

> Build and scale production AI systems while researching and experimenting with novel modeling ideas. The role requires strong software engineering, Python and ML framework proficiency, GPU kernel development, distributed training experience, and familiarity with Transformer-based sequence models.

## Job Description

## Responsibilities
- Design, build, and scale AI systems for serving users.
- Research, implement, and experiment with ideas on supercompute and data infrastructure.
- Write production code and conduct research based on individual interests and organizational needs.
- Learn from and collaborate with leading researchers.

## Requirements
- Extremely strong software engineering skills.
- Proficiency in Python and machine-learning frameworks and tools 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.

## Nice to Have
- Papers published at top-tier venues such as NeurIPS, ICML, ICLR, AIStats, MLSys, JMLR, AAAI, Nature, COLING, ACL, or EMNLP.

## Compensation and Benefits
- Weekly lunch stipend of $75/£75 or equivalent in local currency.
- Full health and dental benefits, including a separate mental-health budget.
- RRSP matching, 401(k), or pension scheme.
- 100% parental-leave top-up for up to six months for either parent.
- Annual enrichment benefits covering arts and culture, fitness and wellness, quality time, and workspace improvements.
- Education and learning stipend for conferences, courses, and coaching.
- Six weeks of paid vacation.
- Travel budget for remote employees visiting other offices and an annual company offsite.
- Co-working benefit for employees not near an office.
- $500 home-office stipend.

## Similar roles

- [Staff Machine Learning Engineer, Traffic Intelligence](https://hotfix.jobs/jobs/cee28c3a-eb64-42f7-a584-8d2b4a0980b8) - Airbnb - Remote - $212k – $265k/yr
- [Member of Technical Staff, Agentic Environments](https://hotfix.jobs/jobs/f3ceb248-2d3e-4b28-a5d1-10891969916e) - Cohere - Remote
- [Staff Machine Learning Infrastructure Engineer, Embedding Platform](https://hotfix.jobs/jobs/9b4387bf-ac16-48ba-810b-2e6a612e3617) - Reddit - Remote - $253k – $355k/yr
- [Senior Staff Machine Learning Engineer, Feed Relevance](https://hotfix.jobs/jobs/a2f9b178-c6cf-49bf-ba93-00688d1027fb) - Reddit - Remote - $266k – $372k/yr
- [Staff Applied Scientist](https://hotfix.jobs/jobs/d518c7a7-07d4-4f06-af79-9c478df769ca) - Garner Health - New York, NY - $300k – $390k/yr

**Apply:** https://hotfix.jobs/jobs/32a9b4f0-a661-4188-a9f2-05553e05c937
**Canonical:** https://hotfix.jobs/jobs/32a9b4f0-a661-4188-a9f2-05553e05c937