Machine Learning Engineer Jobs

Snorkel AI

Snorkel AI is seeking an early member for its ML & Research Engineering team to help build the data and environments behind high-performing frontier and agentic AI. You will be responsible for making the process of generating and evaluating frontier AI data faster, cheaper, and more rigorous using ML and AI. This role involves studying data generation and evaluation, forming and validating hypotheses with production data, and shipping scalable solutions. You will play a key part in shaping the direction, standards, and growth of the discipline.

$208k - $315k

SambaNova

SambaNova is seeking a Principal Place and Route (PnR) Engineer to join our team. This role involves developing and tuning the software that maps dataflow graphs onto SambaNova's Reconfigurable Dataflow Unit (RDU). You will work on partitioning, placement, and routing algorithms to optimize the mapping of customer workloads, including large language model inference and training, onto our hardware. Your contributions will directly impact the throughput and performance of compiled programs and will be integrated into production releases.

$0k - $1000k

Databricks

Databricks is seeking an engineer to join a new, nimble team in NYC to build CustomerLake, the Customer Data Platform on Databricks. This platform aims to bring enterprise-grade ML and AI personalization to companies whose data already resides on Databricks, enabling them to power personalization, recommendations, churn, and lifetime-value modeling without an in-house ML/AI team. This is a unique opportunity to work in a 0-to-1 environment, combining startup excitement with the resources of a major tech company.

$190k - $253k

Stripe

Financial Connections is Stripe's open banking platform, enabling businesses to securely access consumer-permissioned financial data. Our platform connects to thousands of financial institutions, powering use cases from account verification to risk assessment to personal financial management. Across the Financial Connections Engineering org, we focus on delivering high-quality, enriched bank data at scale — building the ML systems that transform raw financial data into actionable signals for both internal Stripe teams and external merchants. Our ML work spans transaction categorization, risk scoring, data enrichment, and the development of intelligent systems that improve data quality across our network. We operate at the intersection of fintech infrastructure and applied machine learning, solving problems that directly impact Stripe's ability to serve millions of businesses and consumers.

Salary not listed

As a Senior AI Performance Engineer, you will analyze and optimize AI training and inference workloads across large-scale distributed systems. Your work will connect compute, memory, communication, and software behavior, owning complex performance investigations from problem definition through validation. You will translate profiling, benchmarking, and modeling results into actionable improvements for engineers. This role offers broad scope across hardware, software, networking, and system architecture, contributing to Graphcore's delivery of efficient and reliable AI systems for large-scale deployment.

Salary not listed

NVIDIA is seeking a Software Engineer, Performance Instrumentation to join their team in Santa Clara, CA. This role offers an exciting opportunity to contribute to groundbreaking projects at the intersection of AI and accelerated computing. You will be instrumental in defining and maintaining performance data requirements, identifying and addressing visibility gaps, and ensuring the quality and efficiency of instrumentation APIs. This position requires a strong background in software engineering with a focus on performance and data quality within machine learning or systems software environments.

$168k - $270k

Stripe

Stripe is seeking a Machine Learning Engineer for its Link Fraud and Auth team. This role involves building and operating models and risk decisioning systems to protect Link while enabling legitimate payments. You will manage the full machine learning lifecycle, from analyzing fraud patterns and identifying opportunities to developing, deploying, monitoring, and improving models in production. Your work will directly impact Link's fraud performance, authorization rates, and expansion into new products and payment experiences, contributing to Stripe's mission to increase the GDP of the internet.

Salary not listed

Stripe

Growth Platform at Stripe builds machine learning systems to help businesses discover and utilize Stripe products. These recommendations are integrated across various platforms including the Dashboard, email, onboarding, documentation, and AI agent interfaces. The team combines business understanding with models to determine the most useful recommendations, optimal timing, and effective learning from outcomes. Their work encompasses recommendation and ranking models, contextual bandits, agent-based recommendations, and the underlying data and evaluation systems, aiming to drive business growth through relevant product adoption while maintaining a positive user experience.

Salary not listed

Stripe

Stripe is seeking a Backend Engineer for its Core Technology team. This role involves working on critical tier0 infrastructure systems, including compute, networking, document databases, distributed caching, and high-assurance engineering. The team builds the foundational platform that enables Stripe products and services to operate at scale, driving reliability, availability, efficiency, and scalability. You will design, build, test, and ship high-quality code daily, contributing to a great customer experience for both internal and external Stripe teams.

Salary not listed

Stripe

The Radar ML team builds the fraud detection models that protect Stripe's $1.9 trillion payment network from fraud. The team owns over 10 real-time deep learning models that must constantly evolve to stay ahead of fraudsters. Each ML improvement translates directly into dollar impact for Stripe and its users. The team's models also power the Radar product suite that tens of thousands of businesses use to screen payments and manage fraud. In this role, you will own ML work across the full lifecycle: researching new fraud patterns, building and deploying models, and sharing results directly with top Stripe customers. You will have opportunities to optimize Stripe’s most intensive ML models, and opportunities to ship 0-to-1 products from scratch.

Salary not listed

As a Senior Machine Learning Engineer in the Applied AI team, you will contribute to advancing AI technology by developing and optimizing new and existing AI models for specialized hardware. You will work on large-scale systems where performance is critical to project success, with visibility across the entire pipeline from hardware to AI applications. This role involves close collaboration with researchers in a rapidly evolving field, requiring continuous learning and adaptation to new challenges. We seek engineers with strong technical foundations who are curious and eager to understand and advance AI model implementation at scale. This position offers flexibility through a hybrid working model from our UK offices.

Salary not listed

NVIDIA is seeking a Developer Technology Engineer specializing in AI to work directly with key application developers. The role involves understanding their challenges and optimizing solutions using GPUs through reference code and direct contributions. You will collaborate with various NVIDIA teams to influence the design of future architectures and software platforms by analyzing application performance and developer efficiency. Occasional travel for conferences and on-site developer visits is expected.

Salary not listed

Accenture

As an AI/ML Computational Scientist, you will design, build, and operationalize artificial intelligence and machine learning solutions for enterprise clients. This involves combining custom models with cloud and third-party AI services to deliver production-ready outcomes. Your role will cover the entire solution lifecycle, from assessing client needs and data, to selecting and customizing models (including Deep Learning, Generative AI, and Large Language Models). You will also design scalable data and MLOps pipelines for training and production, ensuring the quality, value, and reliability of deployed systems.

Salary not listed

Databricks

As a Staff Data & AI Technical Solutions Engineer, you will be responsible for driving and mentoring others in producing Data & AI technical solutions for customer-reported issues. This involves deep dives into production data pipelines, AI workflows, and streaming data ingestion. You will leverage your deep expertise in Data & AI architectures to maintain complex production environments and adhere to SLAs. Collaboration with product engineering and customer account teams is key to delivering results, especially during critical production downtime. You will provide customers with guidance and expertise to build performant architectures and reliable services using Databricks' platform, acting as a technical expert within a global support engineering organization.

Salary not listed

We are seeking a Senior Developer Technology Engineer specializing in Artificial Intelligence. This role involves researching and developing techniques to accelerate AI workloads, particularly deep learning and machine learning, on advanced computer architectures. You will identify and eliminate system bottlenecks to achieve optimal performance on pioneering hardware. A key aspect of this position is collaborating with other technical experts and the developer community, publishing findings, and influencing the design of next-generation hardware and software. This is an opportunity to work at the forefront of technology breakthroughs with an industry leader.

Salary not listed

Join Graphcore's Graduate Programme for a two-year, hands-on learning experience in AI and machine learning test systems. You will apply your software skills to test how AI and ML workloads perform on Graphcore's accelerator hardware. This role involves creating representative ML workloads, analyzing their numerical accuracy and performance, and designing tests to improve testing strategies and infrastructure across simulated and physical hardware. You will also build CI/CD pipelines, troubleshoot complex issues, and translate findings into measurable improvements, bridging the gap between ML frameworks and hardware to enhance software quality and customer experience.

From £45k

Join Graphcore's Graduate Programme for a two-year, hands-on learning experience in AI Compute Libraries & Performance. You will contribute to C++ libraries and compute kernels for machine learning workloads, focusing on developing and optimizing operations, understanding performance bottlenecks, and ensuring the correctness and reliability of software that runs AI models on next-generation Graphcore hardware. This role offers a supportive team environment, mentorship, and opportunities to develop practical experience connecting numerical computing with hardware implementation.

From £45k

Join Graphcore's Graduate Programme for a two-year, hands-on learning experience in a permanent role. You will be trusted to take ownership, share ideas, and contribute to shaping innovative AI systems. This program offers the opportunity to tackle challenging problems alongside a supportive team dedicated to your development and success. You will gain exposure to cutting-edge AI models and accelerator systems, working across the full stack from hardware to applications, identifying opportunities to innovate and differentiate Graphcore's technology.

From £45k

Glean is seeking Machine Learning Engineers to join the Enterprise Brain team, focusing on building proactive AI products that revolutionize enterprise workflows by automating tasks and enhancing user productivity. This role involves leveraging deep user understanding and state-of-the-art enterprise graphs, utilizing LLMs, advanced ML techniques, agent orchestration, and cutting-edge ranking methods. You will contribute to a secure, open platform that redefines how enterprises find, use, and act on knowledge, with a focus on enterprise trust and scalability.

$200k - $280k

Moveworks is seeking a Senior Staff ML Engineer to join our AI Safeguards team. In this role, you will be instrumental in designing, building, and shipping critical safeguards within our agent harness. Your work will focus on protecting against prompt injection, single and multi-turn jailbreaks, toxicity, sensitive topics, and sensitive data exposure. You will develop robust evaluation frameworks to measure the efficacy of these safeguards, addressing both false positives and false negatives using diverse datasets. Collaboration with our Security team will be key, translating their findings into actionable eval cases and fixes, and co-developing mitigations. You will also select, integrate, and fine-tune detection models within a shared stack serving both commercial and government environments, while carefully balancing quality, latency, and cost for controls operating in the request path. Additionally, you will build configurable safeguard policies to empower customers in managing sensitive topics and monitor safeguard behavior in production, addressing any regressions.

Salary not listed

MatX is building the compute platform for AGI, developing vertically integrated full-stack solutions from silicon to systems. We focus on hardware and software to train and run the largest ML workloads for AGI, primarily using the Rust programming language. This role involves designing and optimizing kernels that interface directly with our hardware, working in partnership with ML Research and Hardware Engineering teams, and providing expertise on hardware architecture from a programmer's perspective to ensure seamless integration with the software stack.

$160k - $275k

Figure is an AI robotics company developing autonomous general-purpose humanoid robots with the goal of shipping robots with human-level intelligence for home and commercial markets. The Helix team is responsible for developing the core AI systems that power humanoid autonomy. We are seeking a Helix AI Engineer, Video Pretraining to lead the development of large-scale video foundation models trained on diverse real-world and robot-collected data. This role focuses on pretraining models that learn from raw video, capturing motion, interaction, and temporal structure to enable downstream capabilities in perception, prediction, and embodied reasoning.

$200k - $400k

Sift

As a Machine Learning Engineer at Sift, you will bridge the gap between data science and large-scale distributed systems. You will build end-to-end pipelines that extract signals, train custom models per merchant, and serve predictions at production scale with low latency. You will work on an automated machine learning ecosystem that dynamically recalibrates models based on streaming global telemetry data.

$170k - $240k

Machinify is a leading healthcare intelligence company focused on delivering value, transparency, and efficiency to health plan clients. We are seeking a hands-on, technically deep Sr/Director of Engineering to lead our ML Engineering team. This role involves managing and growing a team responsible for building and operating the company's core AI/ML platform at massive scale. You will also act as a key thought partner to the Chief AI Officer, translating data science proof-of-concepts into scalable, production-grade services. The ideal candidate possesses a blend of AI/ML engineering expertise and applied data science fluency, understanding the intricacies of productizing AI/ML capabilities including serving, observability, reliability, and scale.

Salary not listed