Skip to content

Staff AI/ML Engineer

Build and productionize AI/ML systems including recommendations, NLP interfaces, and agentic workflows. Requires 10+ years building production-grade AI/ML systems and experience with foundation models.

240k – 270kNew York, NYML EngineeringOnsite10+ YOE

About the role

What You’ll Do

  • Partner with product, design, and engineering teams to identify high-impact AI/ML opportunities
  • Prototype and productionize AI systems that feel intuitive but do a lot under the hood—recommendations, natural language interfaces, agentic workflows, and more
  • Develop and scale AI/ML infrastructure that powers both internal tooling and customer-facing features
  • Tackle novel UX problems at the intersection of AI, BI, and apps

What You Bring

  • Bachelor’s degree in Computer Science, Engineering, Mathematics, or a related field (required)
  • 10+ years of experience building and deploying production-grade AI/ML systems
  • Deep knowledge of machine learning, deep learning, and applied AI
  • Experience across the full ML lifecycle: data curation, training, deployment, monitoring
  • A track record of building things that ship—whether it’s recommendations, search, machine translation, or something equally complex
  • Experience adapting or training foundation models (language or multimodal) for novel domains

Bonus Points

  • You've built agents that can plan, reason, and use tools
  • You know your way around cloud infrastructure (AWS, GCP, Azure)
  • You’ve worked in a fast-moving startup or high-growth environment

Compensation & Benefits

  • Base salary range: $240k - $270k annually
  • Eligible for stock options
  • Comprehensive benefits package including generous health benefits, flexible time off, paid bonding time, 401k, commuter and FSA benefits, lunch program, dog friendly office

Skills

Machine LearningDeep LearningAi/Ml InfrastructureFoundation ModelsNatural Language ProcessingAWSGCPAzureAgentic WorkflowsRecommendations Systems

Similar roles

ML Engineering jobs

Staff Software Engineer

Build and lead Traba's agentic platform as a founding member of the Agents team. Architect orchestration, evals, model strategy, and integrations for autonomous AI agents in industrial supply chain workflows. Requires 7+ years engineering experience including 2+ years production LLM/agent systems, plus customer immersion and 0-to-1 leadership.

240k – 300kNew York, NY +1ML EngineeringHybrid7+ YOELLMsAPIs

Staff Software Engineer

Build and scale Traba's applied AI platform by integrating frontier models and agents into production systems for automating industrial staffing pipelines. Requires 7+ years deep full-stack experience with TypeScript/Node.js or Python, PostgreSQL, messaging systems, and distributed systems; AI/LLM production experience a plus.

240k – 300kNew York, NY +1ML EngineeringOn-site7+ YOEAIAPIs

Staff AI/ML Engineer

Build and productionize AI/ML systems including recommendations, NLP interfaces, and agentic workflows. Requires 10+ years building production-grade AI/ML systems and experience with foundation models.

240k – 270kSan Francisco, CAML EngineeringOn-site10+ YOEAWSGCP

Staff Applied AI Engineer, AI Experiences & Agents

Build and ship production-grade AI agents and experiences that help creators personalize, automate, and grow. Requires strong product engineering skills, hands-on LLM/agent experience, and rigorous evaluation practices.

240k – 250kLos Angeles, CA +1ML EngineeringOn-siteRAGLLMs

Staff AI Engineer

Staff AI Engineer shapes technical direction of AI agents for audit workflows, owning architecture, LLM infrastructure, and reliability systems. Requires 8+ years experience with deep expertise in TypeScript/Python, RAG, and enterprise-scale LLM products.

240k – 310kSan Francisco, CAML EngineeringHybrid8+ YOERAGLLMs