Remote Jobs RockRemote Jobs Rock

Staff Machine Learning Engineer

πŸ•’ 2 days ago
Machine LearningLarge Language ModelsAgent SystemsTransformers
Apply Now

πŸ“œ Description

  • Design multi-step LLM systems for reasoning, retrieval, planning, and tool use.
  • Own evaluation across offline benchmarks, online experiments, and model-graded review.
  • Identify and fix failure modes such as hallucination and tool misuse.
  • Set quality, latency, and cost targets for the team.
  • Mentor senior engineers and shape how the group works.

πŸ› οΈ Requirements

  • 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

✨ Benefits

  • Equity
Full 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.

Coinbase

Staff Machine Learning Engineer, Compliance Automation (Platform)

πŸ•’ yesterday
CoinbaseπŸ‘₯ 10,000+ employees🏒 Financial Services

Architect production AI agents that automate compliance workflows, owning model strategy, safeguards, and evaluation while partnering with regulatory stakeholders to deliver accurate, auditable decisions.

Machine LearningPythonLLM Application DevelopmentAgentic AI
Affirm

Staff Machine Learning Engineer (Credit International ML)

πŸ•’ 2 days ago
AffirmπŸ‘₯ 10,000+ employees🏒 Financial Services

Lead the technical strategy and development of production-grade machine learning models for international credit decisions, predicting repayment behavior and scaling modeling capabilities across markets.

PythonMachine LearningTabular ClassificationLightgbm
Pinterest

Staff Machine Learning Engineer

πŸ•’ 16 days ago
PinterestπŸ‘₯ 5001 - 10,000 employees🏒 Computer Software

Drive research, architecture, and production deployment of generative recommendation and reinforcement learning systems, turning novel machine learning research into scalable product impact across Pinterest.

Machine LearningArtificial IntelligenceRecommendation SystemsGenerative Recommendation Systems
Plaid

Staff Machine Learning Engineer (Research Scientist) - DFAI

πŸ•’ 9 days ago
PlaidπŸ‘₯ 10,000+ employees🏒 Software Development

Lead the technical strategy and end-to-end development of foundation models, from data curation and pretraining through production serving, evaluation, and monitoring, while guiding teams across Plaid.

Machine LearningFoundation ModelsTransformersLLMs
Faculty

Lead Machine Learning Engineer (Safety)

πŸ•’ 9 days ago
FacultyπŸ‘₯ 501 - 1000 employees🏒 Internet

Leads the strategy, architecture, and deployment of secure, scalable machine learning systems, advancing AI safety evaluations and multi-agent tooling while advising clients and mentoring engineering teams.

Machine LearningLarge Language ModelsAI Safety EvaluationMulti-agent Systems

Trusted by Remote Workers