Applied ML Staff Engineer: Lead Cross‑Functional Projects

Cohere; United States; Remote; USD 180,000 - 250,000; Full time

Benefits offered by this job

Lunch stipend

Health and dental benefits

Job summary

Cohere is seeking a Member of Technical Staff for our Applied ML team. You will collaborate with customers to design and implement end-to-end ML solutions using large language models, driving rapid AI adoption in enterprise contexts.

You will lead cross-functional initiatives, mentor engineers, and shape product direction while partnering with product, legal, and partner organizations. This role combines research excellence with production impact.

Qualifications

• Prior experience in leading ML teams in applied research and production.
• Experience setting and executing technical direction to build or refine products.
• Experience working cross-discipline in matrixed organizations.
• Proficiency in Python and ML frameworks such as TensorFlow, TF-Serving, JAX, and XLA/MLIR.
• Experience using large-scale distributed training and inference.
• Strong mentorship, communication, and problem-solving skills.
• Demonstrated passion for applied ML models and products.
• Bonus: experience working for B2B companies in ML.

Responsibilities

• Plan and execute large-group projects that carry through from ideation to production.
• Bring cross-functional alignment across engineering, product and other disciplines.
• Mentor a distributed team of engineers in subject matter expertise.
• Identify opportunities and gaps in existing models and strategize what to work on.
• Work closely with product teams to develop solutions.
• Engage in collaborations with our partner organizations.
• Assist our legal teams with preparation of patents on developed IP.

Skills

Leading ML teams

Technical direction

Cross-disciplinary work

Python

TensorFlow

TF-Serving

JAX

XLA/MLIR

Distributed training/inference

Mentorship

Communication

Problem solving

Applied ML

B2B ML (bonus)

Tools

TensorFlow

TF-Serving

JAX

XLA/MLIR

Job description

Cohere is seeking a Member of Technical Staff for our Applied ML team. You will collaborate with customers to design and implement end-to-end ML solutions using large language models, driving rapid AI adoption in enterprise contexts.

You will lead cross-functional initiatives, mentor engineers, and shape product direction while partnering with product, legal, and partner organizations. This role combines research excellence with production impact.