Source Job

$161,000–$221,500/yr
United States

  • Drive AI platform architecture and execute the long-term roadmap for machine learning and generative AI.
  • Lead AI infrastructure vision including end-to-end training, fine-tuning, and low-latency inference platforms.
  • Set engineering excellence standards for the full AI/ML SDLC and mentor senior engineers.

Python Docker Kubernetes AWS RESTful API

20 jobs similar to Staff ML Engineer (ML/AI)

Jobs ranked by similarity.

$143,000–$197,000/yr
United States

  • Build machine learning and generative AI platforms used across the company.
  • Create backend infrastructure for scalable, resilient, and high-performance solutions.
  • Deploy and manage applications in production, coding daily with Python and modern frameworks.

Lyra Health provides evidence-based mental health care, serving over 20 million people globally. The company employs a large workforce and emphasizes a collaborative, mission-driven culture focused on social impact and clinical excellence.

US

  • Evolving the AI knowledge platform with retrieval, indexing, and synthesis for organization-wide use.
  • Architecting and operating agentic infrastructure on AWS with cost guardrails and observability.
  • Partnering with product engineering to define the AI platform API surface and building reference agent implementations.

ShiftKey is a healthcare workforce marketplace that connects facilities with licensed professionals to fill shifts, addressing staffing shortages. The company fosters an inclusive and collaborative culture, valuing diverse perspectives.

$153,000–$180,000/yr
US

  • Lead end-to-end development of advanced AI solutions including agentic systems, RAG pipelines, and multi-agent workflows.
  • Architect AI infrastructure with guardrails, access controls, and compliance protocols for ethical deployment.
  • Guide multi-disciplinary teams, setting technical playbook and best practices.

Nava is a consultancy and public benefit corporation that makes government services simple and effective. Since 2015, federal, state, and local agencies have trusted Nava to solve technology modernization challenges, and they maintain a collaborative, remote-friendly team environment.

$180,000–$210,000/yr
US

  • Contribute to the design and evolution of Expression's Agentic AI platform, defining scalable architectures for enterprise AI applications and services.
  • Architect, deliver, and optimize production-grade LLM services, agent workflows, orchestration layers, and retrieval pipelines.
  • Provide technical leadership and mentorship, guide architecture reviews, and raise the bar for engineering quality across the team.

Expression provides data fusion, analytics, software engineering, and spectrum management solutions to the US DoD and national security community. Founded in 1997 and headquartered in Washington DC, they have a "Perpetual Innovation" culture and were ranked #1 on Washington Technology's Fast 50 in 2018.

India

  • Design and implement scalable agentic solutions for diverse use cases.
  • Build and maintain a centralized AI platform with unified APIs for cross-functional team access.
  • Lead technical discussions, mentor junior engineers, and set the technical vision for the AI platform roadmap.

Cotiviti is a healthcare data analytics company that provides payment accuracy and quality improvement solutions. The company has a large global workforce and emphasizes a collaborative, innovative culture.

United States

  • Design, build, and maintain enterprise AI platform capabilities for LLMs, AI agents, RAG, and GenAI applications.
  • Develop reusable AI harnesses and evaluation frameworks to automate testing, model benchmarking, and quality assurance.
  • Implement observability, LLMOps pipelines, and scalable backend services using cloud-native and containerized architectures.

LTS provides technology solutions supporting high-visibility federal missions. The company values innovation, growth, and collaboration.

$180,000–$200,000/yr
US

  • Design and build the company's AI-first platform architecture.
  • Develop agent-based systems, orchestration layers, and inference pipelines.
  • Build scalable data ingestion, transformation, and feature engineering pipelines.

Our client is an early-stage healthcare AI company leveraging AI, data science, and clinical expertise to generate actionable insights from health data. The small team works alongside experienced founders and world-class advisors, building an AI-native platform from scratch.

$130,693–$180,323/yr
Canada

  • Lead the design and implementation of complex ML/AI systems end-to-end within client environments.
  • Embed directly with client teams, translating business problems into AI system designs.
  • Define AI architecture standards and mentor engineers to drive adoption.

Robots & Pencils designs AI systems for a human world, pairing engineering with creativity. The company values ownership and craft, with a team that ships production-ready AI in 30 to 45 days.

$200,000–$220,000/yr
US Unlimited PTO

  • Set technical direction across multiple services and teams as a senior individual contributor.
  • Define AI platform patterns including agent orchestration, tooling, governance hooks, and evaluation surfaces.
  • Stay hands-on, designing and building production systems in Python across the AI stack.

Mitratech builds world-class products that simplify operations in Legal, Risk, Compliance, and HR functions. Serving 20,000 client companies globally, including 30% of the Fortune 500, the company fosters a diverse, inclusive culture that blends entrepreneurial spirit with enterprise investment.

US Unlimited PTO 14w maternity 14w paternity

  • Work directly with customers to understand their workflows and operational constraints.
  • Build and deploy AI-enabled applications, agents, and workflow automation using LLM techniques.
  • Own delivery from prototype to production, including architecture, backend services, and integrations.

Trase Systems provides an end-to-end platform for deploying and optimizing AI in the enterprise, focusing on bridging the 'last mile' of AI adoption. As a startup founded in 2023, the team is small, agile, and collaborative, with a strong emphasis on innovation and direct customer impact.

$143,000–$197,000/yr
US

  • Be part of a team building backend services for core ML/AI product features
  • Create backend infrastructure for scalable, resilient, and performant solutions
  • Deploy and manage various applications in production

Lyra Health is a leading provider of evidence-based mental health care, serving over 20 million people globally. The company has delivered over 15 million therapy sessions and published 35+ peer-reviewed studies, with a culture of collaboration and social impact.

Global

  • Design, build, and deploy production ML and LLM-based systems (RAG, agentic workflows, fine-tuning, embeddings) for enterprise clients.
  • Own technical delivery end-to-end: from architecture and prototyping to deployment, monitoring, and iteration.
  • Mentor and support other ML engineers on the team with code reviews, technical guidance, and knowledge sharing.

TensorOps is a boutique AI consultancy that bridges strategy and execution, designing and shipping production-grade AI systems for enterprise clients. We are a 100% remote team of 11+ people, partnering with unicorns and NASDAQ-listed companies, and have a culture of autonomy, open communication, and continuous learning.

$159,780–$238,080/yr
United States

  • Design, build, deploy, and maintain production LLM-based solutions and agent workflows.
  • Lead high-complexity, cross-functional AI initiatives from ambiguous problem definition through production adoption.
  • Partner with Cybersecurity and Compliance to design PHI-safe data handling patterns and ensure AI solutions align with relevant controls.

Included Health is a healthcare company delivering integrated virtual care and navigation, breaking down barriers to provide high-quality care for every person. They have a remote-first culture and are an Equal Opportunity Employer.

US Unlimited PTO

  • Architect and build AI platforms, working across distributed teams to deliver GenAI products like Abacus Agents, MCP, and RAG systems.
  • Develop ML/AI pipelines for data preprocessing, model training, and evaluation, enabling rapid experimentation and self-healing systems.
  • Mentor software developers, participate in architectural discussions, and communicate technical direction to align teams and stakeholders.

Abacus Insights transforms healthcare data for health plans to improve outcomes, reduce waste, and enhance member and provider experiences. Backed by $100M from top investors, the company is a collaborative, curiosity-driven team of bold individuals embracing AI and automation to innovate in the healthcare industry.

UK

  • Build Production-Grade Systems: Design and implement scalable Python services powering AI-driven workflows.
  • Own Delivery End-to-End: Take ownership of features from design through to production.
  • Shape AI Agentic Systems: Build intelligent workflows using LLMs and agentic architectures.

We build production-ready AI agentic systems that reshape how financial institutions operate. We are a rapidly growing FinTech backed by major investors, recognized as Fintech of the Year for two consecutive years, with a team of experienced engineers and PhD-level AI specialists.

US

  • Lead the design and delivery of complex enterprise-grade generative and agentic AI systems on AWS using services like Amazon Bedrock and SageMaker.
  • Architect scalable, secure, and production-ready AI platforms aligned with the AWS Well-Architected Framework.
  • Mentor engineers and collaborate with cross-functional teams to drive AWS AI practice evolution.

Robots & Pencils builds smart systems for a human world, blending creativity, engineering, and AWS-powered AI to help organizations reimagine how they work. They are a team of engineers and creatives with a culture of innovation and craftsmanship, focused on delivering enterprise-scale solutions.

Global

  • Design and build the AI execution platform with event-triggered workflows and model-agnostic runtimes.
  • Develop evaluation layers with golden test suites and safety checks to ensure AI reliability.
  • Implement governance mechanisms and build AI agents supporting business workflows.

Jobgether is a platform that connects talent with opportunities using AI-powered matching. The company has a globally distributed team and a remote-first culture.

Latin America

  • Design and own the playbook for AI productization, governance, and reusable internal tooling.
  • Champion AI-assisted development, TDD, and modern CI/CD practices across the organization.
  • Act as the primary technical authority bridging AI/ML architecture and executive strategy.

RYZ Labs is a startup studio that creates industry-defining companies by leveraging top talent and cutting-edge cloud technologies. Founded in 2021, we have a remote team across the US and Latam, with a culture of autonomy, ownership, and continuous improvement.

$120,000–$140,000/yr
US

  • Deploy and configure our AI voice agent for new customers, reducing onboarding effort.
  • Build evaluation suites to validate behavior, catch regressions, and tune prompts.
  • Develop new features and fixes to optimize latency, inference cost, and hallucination guardrails.

PerfectServe is a leading provider of clinical communication and physician scheduling solutions in the health IT space. With over 400 employees, 30,000+ customers, and $100 million+ in annual revenue, the company fosters a collaborative culture that values innovation and cutting-edge product development.

$150,500–$173,000/yr
US

  • Design, deploy, and maintain scalable ML infrastructure for model training, batch processing, and real-time inference.
  • Build and manage cloud-based infrastructure with AWS and Snowflake using Infrastructure-as-Code practices.
  • Develop CI/CD pipelines, automation frameworks, and monitoring for ML systems to improve reliability and governance.

Jobgether is an AI-powered job matching platform that connects candidates with hiring companies. They operate with a team-oriented culture and offer remote work flexibility, focusing on efficient, unbiased recruitment.