Machine Learning Jobs

Distyl AI is seeking an AI Engineer to build and operate production AI systems that deliver business value within customer environments. This role is for engineers who excel in ambiguous problem spaces, take ownership of outcomes, and are eager to work directly on AI systems that must perform reliably under enterprise constraints. AI Engineers are hands-on builders who design, implement, deploy, and iterate on end-to-end AI systems in close partnership with customers, subject matter experts, and other Distyl engineers. They translate operational needs into concrete system behavior, build the necessary software and AI workflows, and continuously improve systems through evaluation, feedback, integration, and production iteration. This is a practical role focused on making AI systems work effectively in real-world scenarios with users, data, and accountability for production outcomes.

$180k - $250k

Figma is seeking an Engineering Manager for Design Systems Adherence to lead a team focused on building workflows, tools, and feedback loops that ensure customers' design and code correctly adhere to their design systems when using Figma's Agents. This role involves enhancing context engineering for AI, improving agent architecture, and developing datasets and evaluators to measure adherence. The manager will set strategy, elevate adherence measurement standards, and translate evaluation results into actionable improvements for design quality. This is a high-impact opportunity to influence Figma's Agent capabilities and overall design quality, addressing the critical need for design and code parity and minimizing drift from defined design systems.

$258k - $376k

The Radar ML team builds the machine learning systems that power Stripe’s Radar fraud and abuse product suite. Our models make real-time decisions across a significant share of internet commerce, helping businesses stop fraud, increase authorization, and grow safely. By turning economy-scale datasets into compelling ML products, the team directly improves customer outcomes and drives meaningful revenue for Stripe. The team is growing and expanding into multiple new areas, including building abuse protections across the full customer life-cycle, adaptive and customizable fraud systems for enterprises, and risk intelligence for over 50 global payment methods. You will own major ML systems and initiatives from problem definition to production, finding where ML can make the biggest difference, designing the approach, building and shipping models and the systems around them, and continuously improving them as data, users, and attackers change. Your work may involve protecting businesses from abuse across the customer lifecycle, building adaptive and customizable fraud systems for Stripe's largest users, or extending risk intelligence to over 50 global payment methods. Across all these, you'll use modern approaches such as transformer models over payment data, graph intelligence across consumers, and LLMs to detect emerging attacks. You will have opportunities to ship 0→1 products and improve mission-critical ML models operating at Stripe scale, often driving several efforts in parallel and setting the technical bar for the team.

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

The Radar ML team builds the machine learning systems that power Stripe’s Radar fraud and abuse product suite. Our models make real-time decisions across a significant share of internet commerce, helping businesses stop fraud, increase authorization, and grow safely. By turning economy-scale datasets into compelling ML products, the team directly improves customer outcomes and drives meaningful revenue for Stripe. The team is growing and expanding into multiple new areas, building abuse protections across the full customer life-cycle, adaptive and customizable fraud systems for enterprises, and risk intelligence for over 50 global payment methods. You will own ML problems from conception to production, analyzing new fraud and abuse patterns, experimenting with model architectures, designing evaluations, deploying models, and monitoring performance. Your work may include developing transformer architectures for payment data, building graph-powered intelligence, leveraging LLMs to detect emerging attacks, or rapidly prototyping new protections.

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.

The Imagine team is redefining AI-driven media experiences for Grok users worldwide. As a Product Safety Engineer on the Imagine team, you will build the critical safety systems and infrastructure that ensure Grok’s multimodal generation capabilities (images, video, audio, and beyond) are powerful, delightful, and responsibly deployed. You’ll design and scale safeguards, detection systems, and evaluation frameworks that prevent harm while preserving creativity and user freedom. Your work will directly shape how millions of users experience generative media — balancing rapid innovation with rigorous safety, trustworthiness, and alignment with SpaceXSI's mission. If you’re a driven engineer passionate about AI safety, content integrity, and building robust systems that protect users without stifling innovation, this is your opportunity to make a lasting impact at the frontier of multimodal AI.

$180k - $440k

We are seeking a Senior AI Developer Technology (DevTech) Engineer to accelerate the frontier of generative AI training and inference. Our team bridges the NVIDIA platform with developers worldwide, diving deep into customer projects to eliminate bottlenecks and partnering with the developer community to deliver optimal performance on NVIDIA systems. If you are passionate about accelerating advanced AI models and influencing the future of NVIDIA hardware and software, we want you on our team.

$184k - $357k

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

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.

Moveworks is seeking a Senior Staff Security Engineer to focus on AI Safeguards. In this role, you will be responsible for maintaining an up-to-date map of ML usage and safeguards across our products, and leading the security design for new ML features. You will actively test the boundaries of our safeguards against various attacks like jailbreaks, prompt injection, and toxicity, translating these findings into evaluation cases that are applied to every model, prompt, and safeguard change. Additionally, you will collaborate with ML engineers to design and build robust mitigations that remain effective even when detection fails, and define and report on key security metrics to demonstrate the efficacy of our safeguards. You will also stay abreast of the latest AI attack research, incorporating new techniques into our testing and mitigation strategies.

The Voyage AI team at MongoDB is building state-of-the-art embedding models and rerankers for efficient unstructured data search and retrieval, crucial for RAG, recommendation systems, and semantic search. This team, comprised of AI researchers from top universities, was recently acquired by MongoDB and is integrating its advanced embedding models with MongoDB's data platform to create powerful end-to-end solutions. We are seeking a Senior Research Scientist to join this dynamic team and contribute to the development of next-generation AI models, tackling complex challenges at the intersection of machine learning research and the practical deployment of large neural networks.

$126k - $248k

The Voyage AI team at MongoDB is building state-of-the-art embedding models and rerankers for accurate and efficient unstructured data search and retrieval, crucial for RAG, recommendation systems, and semantic search. This team, comprised of AI researchers from top universities, was recently acquired by MongoDB and is integrating its advanced embedding models with MongoDB's data platform to create powerful end-to-end solutions. We are seeking a Staff Research Scientist to contribute to the development of next-generation AI models, tackling complex challenges at the intersection of machine learning research and the practical deployment of large neural networks.

$151k - $297k

Glean is seeking recent university graduates with a passion for software engineering to join our team. As a University Graduate Software Engineer, you will participate in designing, developing, and maintaining software solutions that power our products and services. You will own impactful infrastructure problems from inception to production launch, write design documents and robust code, and collaborate with a strong team to prioritize projects. Team placements will be made based on interest and open roles, with opportunities in Backend, Product, and ML Engineering.

$115k - $150k

SambaNova is seeking a new grad ML Infrastructure Engineer to join their team. This role involves writing production Python code for the inference serving stack, including vLLM integration and public APIs. You will contribute to bringing features like speculative decoding, prompt caching, and tool calling into production on the RDU, and develop accuracy and regression tests to ensure quality before customer release.

$131k - $197k

Databricks

The AI Forward Deployed Engineering (AI FDE) team at Databricks is a specialized customer-facing AI team focused on building and productionizing first-of-its-kind AI applications. This role is for experienced engineers who excel at working with customers and teammates, and who are passionate about the latest trends in Generative AI, LLMOps, and Machine Learning. You will collaborate cross-functionally with engineering and product teams to shape strategic priorities and initiatives, acting as a trusted technical advisor to customers across various domains.

$182k - $250k

CoreWeave is seeking a Staff Machine Learning Engineer to join the AI Insights team. This role will be a technical leader responsible for defining and implementing machine learning systems that power anomaly detection, signal correlation, incident understanding, recommendations, and other intelligence capabilities across CoreWeave’s observability and cloud platforms. You will work across the full machine learning lifecycle, from problem framing and model development to evaluation, service building, and scaled operation. The position involves close collaboration with software engineers, product managers, researchers, and domain experts to transform innovative ideas into reliable customer experiences, focusing on building underlying ML systems for AI-powered troubleshooting and optimization in real-world environments.

$180k - $250k

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.

Anthropic

As a senior data science leader for Model Experience, you will play an instrumental role in shaping data-informed decision-making across the organization. You will be a strategic thought partner to product, engineering, and research leaders, influencing model roadmap and launch decisions by providing insights into user experience and model performance beyond benchmark scores. In this tech lead manager role, you will also be responsible for hiring, managing, and developing a team of data scientists, setting their agenda, leading high-stakes analyses, and elevating the impact of data on model and product decisions.

$380k - $460k