Member of Technical Staff, MLE

CohereCohere·Remote(San Francisco)
Other

WFA Digital Insight

The demand for skilled machine learning engineers grew significantly in recent years, with a focus on large language models. As companies like Cohere continue to push the boundaries of AI, the need for professionals who can develop and customize these models has increased. With the remote job market offering more flexibility, candidates can now apply for roles that match their skills without geographical limitations. Cohere stands out for its commitment to scaling intelligence and serving humanity, making this a compelling opportunity for those passionate about AI's potential impact. Before applying, candidates should be prepared to demonstrate their technical expertise and collaborative mindset.

Job Description

About the Role

The Member of Technical Staff, MLE position at Cohere is a unique opportunity to work directly with enterprise customers, pushing the limits of large language models to solve real-world problems. This role combines technical leadership, solution design, and customer-facing technical impact, making it an exciting challenge for experienced machine learning engineers. As part of Cohere's team, you will have the autonomy to define what Applied ML at the company becomes, influencing the capabilities of its foundation models and contributing to the next generation of frontier models.

The day-to-day responsibilities will involve working closely with customers to understand their domains and design custom LLM solutions. This includes rapidly understanding complex business problems, framing them as ML problems, and delivering production-ready models that solve high-value issues. You will also have the opportunity to train and customize frontier models, leveraging Cohere's full stack and contributing improvements back to the foundation model stack.

What You Will Do

  • Contribute to the design and delivery of custom LLM solutions for enterprise customers
  • Translate ambiguous business problems into well-framed ML problems with clear success criteria and evaluation methodologies
  • Build custom models using Cohere's foundation model stack, including CPT recipes and post-training pipelines
  • Develop state-of-the-art modeling techniques that directly enhance model performance for customer use-cases
  • Work as part of Cohere's customer-facing MLE team to identify high-value opportunities
  • Collaborate with researchers, engineers, and designers to push the state of the art in ML
  • Participate in the development of new capabilities, tuning strategies, and evaluation frameworks for foundation models
  • Present technical solutions and results to both internal and external stakeholders
  • Stay up-to-date with the latest advancements in ML research and apply this knowledge to improve model performance and customer solutions

What We Are Looking For

  • Strong ML fundamentals and the ability to frame complex problems as ML solutions
  • Fluency with Python and core ML/LLM frameworks
  • Experience working with large-scale datasets and distributed training or inference pipelines
  • Understanding of LLM architectures, tuning techniques, and evaluation methodologies
  • Demonstrated ability to meaningfully shape LLM performance
  • A broad view of the ML research landscape and a desire to push the state of the art
  • Leadership experience, with the ability to contribute to the growth and development of the team
  • Excellent communication and collaboration skills
  • Humility and strong collaboration instincts
  • A deep conviction that AI should meaningfully empower people and organizations

Nice to Have

  • Experience with specific ML frameworks or tools (e.g., TensorFlow, PyTorch)
  • Knowledge of cloud computing platforms (e.g., AWS, Google Cloud)
  • Familiarity with agile development methodologies
  • Participation in open-source ML projects or contributions to ML research

Benefits and Perks

  • Competitive salary and equity package
  • Opportunity to work on cutting-edge ML projects with a talented team
  • Flexible and remote work arrangements
  • Access to the latest tools and technologies in the field
  • Comprehensive health insurance and benefits package
  • Generous PTO policy and paid holidays
  • Professional development opportunities and support for ongoing education and research
  • Participation in a dynamic and rapidly growing company with a strong mission

How to Stand Out

-

This is a remote position listed on WFA Digital, the platform for professionals who work from anywhere. Browse more remote jobs across all categories.