Source Job

US

  • Own technical direction for high-impact AI products.
  • Work across teams to turn big ideas into shipped systems.
  • Help raise the bar for how we build, evaluate, and operate AI in production.

Python AI/ML LLM

20 jobs similar to Staff AI Engineer

Jobs ranked by similarity.

US

  • Architecting core platform primitives while setting the standard for technical excellence and safety.
  • Designing durable, production-grade systems that power patient engagement and provider workflows at scale.
  • Owning the AI foundation that determines how generative AI shows up across mental healthcare.

Rula is dedicated to treating the whole person, not just the symptoms, and aims to create a world where mental health is embraced as an integral part of one's overall well-being. They are a remote-first company that hires in most U.S. states and are dedicated to having a culture of inclusion.

$194,000–$228,000/yr
US

  • Design, build, and ship LLM-powered features and agentic workflows for Gametime users.
  • Build and maintain evaluation frameworks and prompt testing pipelines for AI-powered experiences.
  • Contribute to orchestration layer, including agent routing, tool use, and multi-step workflow coordination.

Gametime helps people connect through shared live experiences. They operate platforms on iOS, Android, mobile web, and desktop, supporting over 60,000 events across the US and Canada, fostering a collaborative and inclusive environment where diverse perspectives are valued.

LATAM

  • Design and build AI systems in production that solve real business problems, end-to-end: from discovery to operation.
  • Work with product and operations to translate ambiguous problems into measurable and maintainable solutions.
  • Build data pipelines that feed models and product surfaces.

Skydropx is innovating logistics with a team of visionary people who want to grow and change the world. They are integrating AI into their logistics platform for LATAM, working at a multi-tenant scale with hundreds of thousands of shipments per month.

  • Design, develop, test, and deploy AI/ML models and applications including NLP pipelines, predictive models, recommendation engines, and intelligent automation workflows.
  • Build and integrate large language model (LLM) powered features using APIs such as OpenAI, Azure OpenAI, or Anthropic; implement retrieval-augmented generation (RAG) patterns and AI agent workflows.
  • Develop and maintain data pipelines that support model training, fine-tuning, evaluation, and real-time or batch inference.

ExtensisHR is a Professional Employer Organization (PEO) in the U.S. with client employees in all fifty states. They deliver personalized HR services for HR, employee benefits, payroll and taxes, employer risk, compliance, and employee management.

Global

  • Contribute to the development of the Everywhere Inference platform, a Kubernetes-based solution.
  • Design and implement APIs and developer tools to simplify deployment, management, and monitoring of AI applications.
  • Optimize serverless container workflows for AI workloads, ensuring performance, scalability, and seamless autoscaling.

Gcore provides infrastructure and software solutions for AI, cloud, network, and security. They have 550+ professionals globally and power everything from real-time communication and streaming to enterprise AI and secure web applications.

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

  • Design, develop, and deploy AI/ML models to automate and improve internal workflow.
  • Build and maintain ML pipelines within an AWS cloud environment.
  • Integrate ML capabilities into existing Java and React application workflows.

Oddball aims to improve daily lives by delivering quality software to the federal space. With a team of experienced engineering, product, and UX professionals, we value learning, growth, and making a big impact in a rapidly growing company.

US

  • Launch and scale AI-powered features that transform the patient experience.
  • Explore how AI can support patients before, during, and after care with their therapist.
  • Build safe, supportive, and personalized AI experiences that enhance care.

Rula provides mental healthcare with AI-enabled experiences. They aim to enhance the human connection at the core of therapy and are a remote-first company with a culture of diversity, equity, and inclusion that supports employees in feeling safe, seen, heard, and valued.

US

  • Design, build, and maintain production-grade AI systems and customer-facing AI features.
  • Develop agentic workflows using LLMs, retrieval systems, tools, APIs, and backend services.
  • Design and implement retrieval-augmented generation (RAG) systems, including ingestion pipelines, embeddings, semantic retrieval, and context assembly.

Givzey is a fast-growing and innovative technology company serving the nonprofit sector, on a mission to unlock more generosity through AI-powered donor engagement. In just three years, Givzey’s platform has already helped organizations raise $10M+ through autonomous engagement.

$428–$500/mo
Global

  • Own the complete lifecycle of your assigned workstream.
  • Build AI systems designed to perform consistently and efficiently at scale.
  • Collaborate with our team to translate requirements into maintainable AI solutions.

They are looking for a contract AI Engineer to join them for an active, ongoing project and contribute immediately. Contract performance can lead to repeat engagements.

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

  • Build 0-1 AI systems transforming core brokering workflows, from processing complex commercial policies to enabling brokers to manage renewals simultaneously.
  • Embed directly with brokering teams, underwriters, and operations to identify automation opportunities, building solutions side-by-side with users.
  • Ship production-ready AI features delivering measurable impact, reducing quote turnaround time and increasing broker capacity through intelligent market matching.

Newfront, a WTW company, creates agentic services that handle routine insurance work, freeing brokers for high-stakes strategy. They deploy autonomous systems for placement, AI analysis, and workflow orchestration in small, specialized teams embedded with the business.

Global Unlimited PTO

  • Diagnose business problems before building solutions, mapping workflows and confirming AI is the right intervention.
  • Own AI initiatives end-to-end, from stakeholder discovery and technical design through implementation and iteration.
  • Improve organizational flow by building solutions that reduce bottlenecks and increase throughput, measuring success using flow metrics.

GitLab is the intelligent orchestration platform for DevSecOps, enabling organizations to increase developer productivity, improve operational efficiency, reduce security and compliance risk, and accelerate digital transformation. GitLab has more than 50 million registered users and a high-performance culture driven by its values and continuous knowledge exchange.

$90,000–$150,000/yr
US

  • Design and deliver production AI and agentic systems across document intelligence, workflow automation, and copilots.
  • Define architecture decisions for LLM-based systems, including retrieval, tool use, orchestration, memory, and evaluation.
  • Own evals and observability for production AI and manage cost and latency at production volume.

Maxwell is a mortgage technology and fulfillment company with a mission to make lending simpler, faster, and more accessible. They power hundreds of lending institutions with their mortgage Point of Sale and related capabilities and are a remote-first team that takes craft seriously.

$229,000–$240,000/yr
US

  • Lead AI strategy and execution, integrating AI into core products and accelerating development workflows.
  • Architect and build complex, scalable systems and AI-driven features using technologies like Python, Java, AWS, and GCP.
  • Provide technical leadership, mentorship, and cross-functional collaboration to solve complex problems and drive successful outcomes.

Tebra provides an all-in-one EHR+ platform designed exclusively for independent healthcare practices to connect EHR software, billing, automation, telehealth, and marketing. The company serves over 42,000 private practices and fosters a culture of entrepreneurship, customer focus, simplicity, teamwork, and celebration.

AI Engineer

Natera
$105,700–$132,100/yr
US

  • Ship zero-to-one AI products end-to-end — from customer discovery and prototyping through production deployment and iteration
  • Build agentic AI systems — design and implement autonomous and semi-autonomous workflows using LLMs, tool-use, memory, and orchestration
  • Develop AI tools that improve efficiency across clinical operations, data extraction, manual workflows, and more

Natera is a global leader in cell-free DNA (cfDNA) testing, dedicated to oncology, women’s health, and organ health. The Natera team consists of statisticians, geneticists, doctors, scientists, business professionals, software engineers and other professionals from world-class institutions.

AI Engineer

LMI
$111,426–$192,890/yr
US

  • Design and develop data pipelines, scoring algorithms, and API infrastructure to power AI-driven matching and recommendation capabilities.
  • Build and maintain integrations between the matching engine and an existing program management platform.
  • Collaborate with SMEs to build, test, and refine user-configurable matching logic.

LMI is dedicated to accelerating government impact with innovation and speed, bringing commercial-grade platforms and mission-ready AI to federal agencies. Headquartered in Tysons, Virginia, they are committed to delivering impactful results that strengthen missions and drive lasting value.

$83,000–$115,000/yr
US 24w maternity 12w paternity

  • Design, build, and maintain AI-powered workflows and internal tools that support teams.
  • Develop and manage LLM-based solutions tailored to specific operational needs.
  • Collaborate with leads, analysts, and managers to translate business problems into AI solutions.

Chime is a financial technology company that aims to make banking services helpful, easy, and free. They empower members to take control of their finances through user-friendly tools. Chime is a team of problem solvers, dreamers, and builders passionate about helping millions unlock their financial potential.

$215,000–$230,000/yr
US 4w PTO 12w maternity 12w paternity

  • Develop advanced artificial intelligence solutions to improve health outcomes.
  • Collaborate with engineering and analytics teams to integrate AI technologies into existing products and workflows.
  • Train, fine-tune, and use AI models harnessing knowledge from extensive datasets of medical records.

Aledade empowers independent primary care by helping practices, health centers, and clinics deliver better care and thrive in value-based care. Founded in 2014, they've become the largest network of independent primary care in the country with a collaborative, inclusive, and remote-first culture.

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

  • Evaluate and refine AI prototypes built by business units to enhance commercial ROI, security, and architecture.
  • Refactor high-value internal AI prototypes into secure, scalable, enterprise-grade applications.
  • Build and maintain secure LLM integrations with internal systems like data lakes and Salesforce, ensuring full-lifecycle maintenance of applications.

Impiricus is an AI-powered HCP Engagement Engine that ethically connects healthcare professionals to pharmaceutical resources to reduce go-to-market costs and accelerate patient access to treatments. It is a fast-growing company with a unique network of HCPs and advisors, fostering a collaborative and impactful culture where employees can work flexibly.

US

  • Evaluate AI-generated coding interactions end-to-end.
  • Assess quality of explanations and reasoning.
  • Provide constructive feedback on outputs.

Jobgether is a platform that uses AI to match candidates with jobs. They ensure applications are reviewed quickly, objectively, and fairly against the role's core requirements.

$0–$0/yr
US Canada

  • Design, build, and ship agentic workflows across multiple domains.
  • Build multi-step agents capable of autonomous planning, context tracking, memory, tool use, and API orchestration.
  • Drive technical and architectural decisions to meet product requirements while also anticipating and designing for future needs

Cority helps customers see and prevent risks across their operations in real time. Our EHS+ platform converges people, data, and AI agents to provide a clear view of information people can trust. For 40 years, Cority has been the market leader in EHS+, recognized by top analysts and trusted by more than 1,500 of the most complex organizations worldwide.