Back to Jobs
Recently postedRemoteFull timeTechnology

Staff Machine Learning Engineer

ThemuralgroupRemote (US)🇺🇸United States

Quick overview

SeniorityMid Senior
Employment typeFull Time
Work modeRemote
LocationRemote (US), United States
LLMPyTorchGenerative AI

Job description

Location: Remote (US)

Pay: $230,000–$300,000 + bonus + equity

A late-stage, venture-backed enterprise AI company is hiring a Staff ML Engineer to set the technical direction for its agent systems. This is model and systems work that ships to large enterprise customers, not a thin layer over someone else's API.

What you'll do

  • Design multi-step LLM systems: reasoning, retrieval, planning and tool use

  • Own evaluation, from offline benchmarks to online experiments and model-graded review

  • Find and fix failure modes such as hallucination and tool misuse

  • Set the quality, latency and cost targets the team builds against

  • Mentor senior engineers and shape how the group works

What you bring

  • 7+ years shipping ML in production, including LLMs at scale

  • Led the design of complex, multi-step LLM systems

  • Depth in transformers, embeddings and retrieval

  • Evaluation experience beyond single prompts

  • PyTorch or similar, on distributed cloud infrastructure

  • BS in CS or a related field; MS or PhD a plus

Nice to have: ML shipped before LLMs, then moved into generative AI; detailed retrieval work (chunking, reranking, evals)

Mural Group is recruiting for this role on behalf of a client. Apply here and a Mural recruiter will be in touch.