Source Job

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

  • Scope and lead ML initiatives end-to-end from identifying opportunities through production deployment.
  • Design, develop, and optimize ML models and AI systems for document processing and automation.
  • Build and maintain production ML pipelines that are robust, observable, and scalable.

Python ML AI PyTorch LLM

20 jobs similar to Staff Machine Learning Engineer

Jobs ranked by similarity.

Global

  • Architect and build agentic workflows that combine large language models, reasoning components, and data pipelines.
  • Lead the design and development of advanced ML/NLP products, from ideation to production.
  • Drive experimentation with new approaches for agentic reasoning, coordination, and autonomous system design.

SmartRecruiters is the Recruiting AI Company that transforms hiring for the world’s leading enterprises. Built for global scale, SmartRecruiters, an SAP company, delivers an AI-powered hiring platform that automates and optimizes the entire talent acquisition process. They are a values-driven, globally focused tech company with strong financial backing and a bold vision for the future of work and foster a collaborative and inclusive work environment.

Europe Unlimited PTO

  • Contribute to designing, evaluating, and shipping our mental health AI Agent and its supporting infrastructure.
  • Develop and maintain robust data pipelines to power model training and evaluation.
  • Partner with AI Research, Product, and Engineering teams to define new features.

Sword Health is shifting healthcare from human-first to AI-first through its AI Care platform. They aim to make world-class healthcare available anytime, anywhere, while significantly reducing costs. Backed by clinical studies and patents, Sword Health has raised more than $500 million from leading investors.

US

  • Develop, test, and deploy LLM-powered extraction pipelines for clinical text at scale.
  • Automate prompt execution, result validation, and error handling to enhance reliability.
  • Monitor and maintain production AI models, ensuring uptime, accuracy, and compliance.

iCIMS is a software company. The job posting mentions thriving in a start-up environment.

$170,000–$245,000/yr
US

  • Design, prototype and productionize scalable machine learning and optimization models.
  • Develop frameworks, pipelines, libraries, utilities and tools that process massive data for ML tasks.
  • Build end-to-end reusable pipelines from data acquisition to model output delivery.

BetterHelp's mission is to remove the traditional barriers to therapy and make mental health care more accessible. Founded in 2013, they are now the world’s largest online therapy service, with over 30,000 licensed therapists.

Canada Unlimited PTO

  • Lead the design, development, and deployment of production, multi-turn LLM-powered features.
  • Own backend services in Python that integrate LLM agents with Fullscript’s platform and support reliable production use.
  • Partner with medical, product, and engineering teams to identify high-value opportunities for AI and turn them into practical, scalable product capabilities.

Fullscript is a health technology company committed to helping people get better by creating a platform that powers every part of care. More than 125,000 practitioners use Fullscript for clinical insights, lab interpretations, patient analytics, education, and access to high-quality supplements.

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

  • Design and implementation of reliable, maintainable, and scalable GenAI systems.
  • Serve as a subject matter expert for machine learning systems owned by the team.
  • Mentor junior and mid level engineers through code reviews and design collaboration.

Trajector specializes in medical evidence services, guiding clients through disability benefits complexities. They are a global team of over 1,800 dedicated individuals, streamlining the path to benefits and ensuring access to rightful compensation for those with disabilities.

US

  • Build and deploy end-to-end AI/ML solutions, from data pipelines and feature engineering to model training and inference
  • Develop and maintain data pipelines for ingesting, transforming, and preparing data for analytics and machine learning
  • Write clean, modular, and maintainable code to support scalable AI applications

Eimagine fosters a remote-enabled environment where their people can thrive. They are a team of professionals who take pride in their craft, continuously learn, and support one another, helping clients navigate technology and business change while delivering meaningful outcomes.

US

  • Design and Develop machine learning infrastructure, tooling, and models to help teams deliver world class experiences.
  • Help product and development teams understand the data lifecycle and the inherent experimental nature of machine learning.
  • Build internal products and platforms to enable teams to incorporate AI into their features and customer facing products.

Weave provides an all-in-one platform for small businesses to streamline communications, and patient experiences. The company has a phenomenal culture, and Weave's teams are cross-functional agile teams composed of a product owner, backend and frontend devs and devops.

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

  • Own the full ML lifecycle, including data ingestion, training, validation, deployment, monitoring, retraining, and retirement.
  • Transition AI/ML prototypes into scalable, production-ready systems with CI/CD pipelines, automation, and observability.
  • Develop and maintain AI-driven applications and inference services, optimizing for performance, scalability, reliability, and cost.

IMO Health combines software development, artificial intelligence, and clinical expertise to create AI-driven solutions. They enhance access to reliable health information, support clinical decision-making, and improve patient outcomes.

US Unlimited PTO

  • Design, build, and maintain ML infrastructure across training, evaluation, serving, and monitoring
  • Own data pipelines including generation, cleaning, validation, and versioning
  • Build and improve experiment tracking, orchestration, and reproducibility tooling

Quilter is helping electrical engineers save time and accomplish more by automating the tedious and time-consuming task of designing printed circuit boards (PCBs). Their small team is composed of experts in electrical engineering, electromagnetic simulation, ML/AI, and high-performance computing (HPC).

  • Design, implement, and evaluate machine learning models and AI algorithms.
  • Develop and optimize prompts for LLMs to improve model outputs.
  • Collaborate with software engineers, data scientists, and product teams.

Cadre AI is focused on building and optimizing AI-powered platforms, bringing together cutting-edge technologies and expertise in machine learning and large language models. The team is dedicated to advancing AI capabilities and applying them to real-world challenges through scalable, high-impact solutions.

US

  • Design and build the AI platform layer, including data pipelines and serving infrastructure.
  • Productionize AI/ML capabilities, ensuring reliability, performance, and scalability.
  • Architect data pipelines to ingest, transform, and serve data to power AI features.

Lone Wolf Technologies is building AI capabilities into the core of its platform, transforming how real estate professionals manage transactions, serve clients, and grow their businesses. The company's culture fosters innovation and collaboration, empowering employees to contribute to impactful projects.

Europe

  • Own the architecture and delivery of production-grade LLM systems and classical ML solutions.
  • Design, evaluate, and optimize RAG pipelines (retrieval strategy, chunking, indexing, monitoring).
  • Build scalable, production-grade LLM services and agentic workflows, alongside traditional ML systems where appropriate.

Hiflylabs is a team of 250+ data and tech enthusiasts based in Budapest. They focus on data engineering, data science, artificial intelligence and application development, working on a wide range of projects around the world. Hiflylabs values its people and is committed to nurturing their personal and professional development through a mentoring system.

US

  • Conduct end-to-end analyses, wrangling data via SQL or Python, to statistical modeling, to hypothesizing and presenting business ideas.
  • Mentor and guide junior engineers, fostering a culture of learning and technical excellence.
  • Support the development and deployment of projects involving machine learned models for offline, batch-based data products as well as models deployed to online, real-time services.

Yelp's mission is connecting people with great local businesses, requiring the use of cutting-edge Machine Learning (ML) and Artificial Intelligence (AI). Their engineering culture values individual authenticity and encourages creative solutions, supporting growth in a collaborative environment.

US

  • Lead Rula’s applied AI investments as they scale.
  • Own technical direction for high-impact AI products and work across teams to turn big ideas into shipped systems.
  • Help raise the bar for how they build, evaluate, and operate AI in production.

Rula strives to create a world where mental health is no longer stigmatized and provides quality, evidence-based care. They are a remote-first company that is dedicated to treating the whole person, not just the symptoms, and making a positive impact in the field of mental healthcare.

Australia

  • Drive the design and evolution of AI-ready tools and APIs for LLM platforms.
  • Own and evolve evaluation frameworks that measure tool-use accuracy across platforms.
  • Shape Canva's agent architecture, making strategic technical decisions about intelligence location.

Canva is a design platform that enables users to create various visual content. They have offices in multiple locations in Australia and New Zealand, and they offer a flexible work environment.

Europe

  • Build and ship AI-powered product and internal solutions using LLMs, RAG, tool calling, workflows, and agentic patterns
  • Design quality and evaluation frameworks for AI systems, including offline evals, online signals, failure analysis, and continuous improvement loops
  • Contribute to AI platform and tooling decisions that improve reuse, speed, and consistency across teams

Finom is a European tech startup headquartered in Amsterdam, revolutionizing financial landscape for entrepreneurs. They develop an all-in-one financial B2B solution integrating banking, accounting, financial management, and invoicing into a mobile-first platform and nurture innovation in an inspiring work environment.

Canada 5w PTO

  • Build, ship, and own product features end-to-end
  • Work with designers and product managers to create high-performing product features.
  • Apply ML techniques to LLM-based approaches with a strong focus on reliability, performance, and maintainability.

Optro is the leading audit, risk, ESG, and InfoSec platform on the market and has surpassed $300M ARR. They inspire each other to innovate and assist each other to create the most loved platform, which has allowed them to become one of the 500 fastest-growing tech companies in North America.

US Canada

  • Develop and train machine learning models for learned behavior systems.
  • Implement production-quality ML code to support model training, evaluation, and inference.
  • Analyze model performance, identify failure modes, and propose improvements to increase robustness.

Torc Robotics develops behavior models that power decision-making for autonomous trucks. While the job posting does not provide specific information regarding company size, it can be inferred that they foster a highly collaborative environment.

Global

  • Design and develop an AI-powered productivity analytics platform.
  • Build scalable LLM pipelines and create a meta-workflow system.
  • Develop system-level prompt engineering and build an evaluation framework for AI output quality control.

Appflame is a Ukrainian product-driven tech company committed to building world-class products. They have 500+ team members and offices in Kyiv, London, Limassol, and a co-working hub in Warsaw; they value bold, driven people who are passionate about building real products.