Senior Machine Learning Engineer Jobs

NVIDIA is seeking a Senior System Simulation Architect to drive the development of CPU architectures for AI, deep learning, high-performance computing, gaming, virtual reality, and autonomous vehicles. This role involves developing and integrating full-system functional models, bringing up system and application software in simulation and emulation, and supporting CPU architects and performance engineers. The position is crucial for pushing performance boundaries for NVIDIA's CPU products and integrating them seamlessly with NVIDIA's broader technology stack to enable faster AI model training, agentic use-cases, efficient data processing, and scalable cloud deployments.

$184k - $357k

NVIDIA is seeking a Senior MLOps Engineering Manager to lead the build, development, and operation of large-scale, end-to-end data and ML pipelines that power the company's autonomous driving products. This role involves leading a highly technical engineering team responsible for building and operating cloud-scale pipelines that ingest, validate, process, and transform multimodal sensor data into high-quality training, evaluation, and validation datasets. These pipelines are foundational to the autonomous vehicle program and directly enable customer-facing autonomy features. The ideal candidate will be a seasoned engineering leader with strong ownership and a passion for customer-focused development, capable of scaling systems and teams in a fast-paced, multi-functional environment.

$272k - $431k

NVIDIA is seeking a world-class computer architect to contribute to the development of future high-performance computing systems, with a focus on enhancing the power-constrained performance of hardware. The ideal candidate will have a strong track record of understanding and analyzing memory systems architecture to improve performance per watt and performance per millimeter. A broad perspective across computer architecture and depth in power, performance, and area (PPA) analysis is highly desirable. This position offers the opportunity to have a real impact in a fast-moving, technology-focused company.

$272k - $431k

We are the GPU Communications Libraries and Networking team at NVIDIA, responsible for delivering critical communication libraries like NCCL, NVSHMEM, and UCX for Deep Learning and High-Performance Computing (HPC) applications. These applications demand immense compute power and operate at scales involving tens of thousands of GPUs, connected via high-speed interconnects and networking. Communication performance is paramount, directly impacting end-to-end application performance, especially at massive scales. We are seeking a technical leader to manage our NVSHMEM and UCX libraries, offering an exceptional opportunity to advance the state-of-the-art and develop groundbreaking platforms.

$184k - $357k

NVIDIA is seeking a Senior Verification Engineer to join their Display Team, focusing on verifying the world's foremost GPU and SoC display engines. In this role, you will collaborate with Architects and ASIC Engineers to own testplan development, verification sign-off, and infrastructure. The goal is to deliver high-quality IPs that power NVIDIA products across various markets, including consumer, professional, automotive, and AI. You will leverage agentic AI to enhance efficiency in tasks like testplan development, automation, and debug.

$168k - $311k

Step into a role where you can make a significant impact on the future of autonomous vehicles. As a Senior Software Engineer, ML Loop - Automotive at NVIDIA, you will be at the forefront of our data flywheel operation. This is your opportunity to work with hardworking individuals and pioneer innovative technologies to contribute to world-class solutions. Embrace this outstanding opportunity to be part of NVIDIA's ambitious journey and help craft the future of autonomous driving through outstanding data engineering.

$168k - $270k

NVIDIA is seeking a Deep Learning Compiler Engineer to join its Deep Learning Compiler (DLC) team. This role involves developing and optimizing compiler algorithms for deep learning networks, which are crucial for NVIDIA's inference engine across various platforms. The compiler is designed to deliver leading inference performance, fast build times, reduced memory footprints, and ease of use for the global deep learning community. You will collaborate with deep learning software framework and hardware architecture teams to accelerate the next generation of deep learning software, contributing to defining public APIs, performance analysis, and implementing compiler infrastructure techniques.

$152k - $242k

Join a team that analyzes large-scale datacenter workloads on GPU-accelerated clusters. You will transform telemetry and workload data into clear findings and visuals, partnering with OS, container, GPU, and systems engineers. When useful, you will apply machine learning and deep learning techniques for categorization and forecasting, coordinating these into tools the team uses. This role involves analyzing signals to identify opportunities for application and platform improvements, working with high-dimensional data to spot trends, and communicating conclusions to engineers and leadership.

$152k - $242k

NVIDIA is seeking a Senior Software Engineer to join our CSP Engagements team, focusing on system software for Datacenter products such as GB200. This role involves deep technical expertise in embedded firmware, Linux kernel development, and middleware development, combined with customer-facing responsibilities to enable cloud service providers with next-generation computing platforms. You will work at the intersection of hardware and software, driving technical solutions from concept through deployment.

$184k - $357k

NVIDIA is seeking a Senior Systems Software Engineer to join their Deep Learning Solutions team. This role focuses on optimizing deep learning inference for autonomous vehicles and robotics on edge devices. The ideal candidate will be a hands-on specialist capable of analyzing model architectures at the operator level, identifying performance bottlenecks through kernel trace analysis, and evaluating modern AI architectures on GPU and SOC platforms. This position directly contributes to enhancing real-time perception and response capabilities for autonomous systems, working at the intersection of model frameworks, compiler technology, and embedded hardware. Collaboration with automotive OEMs, robotics teams, and internal hardware groups is a key aspect of this role.

$224k - $357k

We are seeking senior systems software engineers to enhance the performance, efficiency, and readiness of the CUDA Driver for next-generation accelerated computing. This role involves developing performance-critical CUDA Driver features and systems software improvements to maximize the potential of NVIDIA GPUs for AI, deep learning, HPC, and other CUDA-powered applications. You will tackle complex performance challenges at the intersection of systems software, CPUs, interconnects, and GPUs, tracing issues from real workloads through the entire stack, designing and shipping production features, and delivering validated optimizations for current and emerging platforms. This position offers the opportunity to combine hands-on engineering with broad technical influence, collaborating with various CUDA software, hardware architecture, frameworks, applications, product, and customer-facing teams. You will lead cross-layer investigations, mentor engineers, and help define performance direction, shaping both current GPU utilization and future CUDA software and GPU architectures.

$184k - $357k

NVIDIA Halos is seeking a Senior Software Safety Engineer to join their team, focusing on the full-stack, comprehensive safety system for Physical AI. This role involves ensuring the safe development and deployment of autonomous machines, AI models, chips, software, tools, and services. You will collaborate with diverse engineering teams across NVIDIA, including those working on autonomous vehicles, robotics, and other safety-critical applications. The position requires guiding teams in specifying, building, implementing, and verifying safety requirements across the software stack, from drivers to end-to-end AI models, and integrating AI safety principles into the Product Life Cycle. Additionally, you will engage in customer interactions and international standardization activities related to AI safety, and support system safety activities like decomposition of safety requirements and architectural safety analysis.

$224k - $357k

NVIDIA is seeking a Backend Compiler Engineer to join our GPU Software organization. The Compiler team is responsible for constructing and emitting high-performance GPU machine instructions for Graphics (OpenGL, Vulkan, DX) and Compute (CUDA, PTX, OpenCL, Fortran, C++). This team comprises leading compiler engineering experts who deliver cutting-edge performance and capabilities for NVIDIA's complex parallel SIMT architectures.

$152k - $242k

NVIDIA is seeking a Senior Machine Learning Engineer to join its prediction team for autonomous driving solutions. This role involves developing and productizing cutting-edge prediction modules responsible for generating multi-modal scene-consistent predictions for road participants. The engineer will focus on enhancing the robustness, accuracy, and efficiency of these solutions to enable autonomous driving in diverse environments and conditions. The position offers the opportunity to work on state-of-the-art ML techniques and contribute to the future of autonomous driving.

$224k - $357k

NVIDIA is building GeForce G-Assist, an on-device AI assistant that integrates Small Language Models (SLMs), retrieval systems, and hybrid cloud capabilities for responsive, context-aware assistance within the GeForce ecosystem. This role involves close collaboration across engineering and product teams to ensure G-Assist performs reliably in real-world scenarios. The focus is on production model behavior rather than just benchmarks, with an emphasis on accuracy, robustness, and conversational reliability. The team works with SLM and VLM architectures, hybrid cloud/local model architectures, and optimizes local inference for performance.

$152k - $288k

Anduril Industries

Anduril Australia is developing un-crewed maritime and air domain systems that leverage unsupervised autonomy for long endurance, multi-mission capability. This Senior Machine Learning Engineer role is crucial for projects like the Extra Large Autonomous Undersea Vehicle (XL-AUV) Program, requiring deep experience in developing, testing, validating, and deploying machine-learning based sensor processing and the associated technical toolchains. The engineer will architect, design, deliver, support, and evolve next-generation autonomous vehicles throughout their product lifecycle, contributing to Anduril's mission of transforming defense capabilities with advanced technology.

Salary not listed

Cloudflare

You will help define how machine learning models run across Cloudflare’s global network, from frontier open LLMs and real-time voice models to customer-deployed models served on heterogeneous GPUs and next-generation accelerators. You’ll work with systems engineers, product teams, hardware partners, and AI/ML engineers to bring models into production with low latency, strong reliability, and efficient resource use. This role combines applied ML, inference optimization, evaluation, and production engineering, with a focus on benchmarking models, improving serving performance, validating quality, and building tooling that helps Cloudflare and its customers ship AI applications at Internet scale.

$194k - $266k

You will be part of a distributed agile team building a new SaaS platform that underpins enterprise applications for data analytics and other complex problems. Your role will involve ensuring the framework and products meet established design, implementation, and quality goals. This position requires a strong backend engineering focus within a globally distributed team, contributing to the architecture, development, and deployment of Securiti's core backend platform.

Salary not listed

Sony Interactive Entertainment (SIE) is seeking a Staff Machine Learning Engineer to join the Visual Computing Group (VCG). This role will focus on advanced R&D to redefine game play and live streaming through cutting-edge neural networks and ML. The VCG team aims to design and deploy systems that push the boundaries of runtime efficiency, visual quality, and latency in game rendering and streaming.

Salary not listed

Cloudflare

We are seeking an experienced Data Engineer to join our Austin team and contribute to the development of our internal data products. In this role, you will build data-driven applications that empower various teams across the company, including go-to-market, engineering, and product. You will be involved in the entire product lifecycle, from defining requirements and designing features to implementation and ongoing ownership. You will focus on creating scalable and reliable systems to address critical business challenges, collaborating with full-stack engineers on new features and managing the data pipelines and services that support our applications. Our technology stack is diverse, encompassing languages like Go and Scala, and technologies such as ClickHouse, with continuous evolution. Many of our products also integrate significant AI components, offering opportunities to develop advanced, AI-ready data layers and work with modern AI infrastructure and tools like vector databases, MCP-based services, and Workers AI.

Salary not listed

Sony Interactive Entertainment is seeking a Senior / Staff Data Scientist to lead the measurement of products, features, and commercial initiatives through advanced experimentation and causal inference. This role involves developing innovative measurement methodologies, defining best practices, and providing technical leadership to enable data-driven decision-making across PlayStation. You will tackle complex measurement challenges beyond traditional A/B testing, applying modern causal inference, statistical modeling, and machine learning techniques to derive robust business insights. Collaboration with product, engineering, analytics, and business teams is key to raising the standard of experimentation and evidence-based decision-making throughout the organization. This is a unique opportunity to influence strategic decisions impacting millions of players and establish best-in-class measurement practices within a leading entertainment company.

Salary not listed

As a Senior Machine Learning Engineer focused on Camera Models, you will be instrumental in developing and deploying machine learning models that power the camera-based perception systems for autonomous trucks. Your work will involve building and maintaining core vision systems that enable the autonomy stack to understand the environment, detect and localize objects, and estimate scene structure from camera data. You will own the end-to-end development of robust, scalable camera-based models, iterating on system performance and delivering production-ready solutions within the autonomy stack, collaborating closely with perception, data, and infrastructure teams to ensure safe and reliable autonomous driving.

CA$168k - CA$193k

We are seeking passionate Machine Learning Engineers to join our team and contribute to AI transformations and scaling production models. In this role, you will leverage cutting-edge open-source frameworks, advanced algorithms, and emerging technologies to solve complex business problems in a fast-paced, collaborative environment. You will have the opportunity to design, build, and deliver ML models, scale multi-tenant platforms for model training and serving, and inform ML infrastructure decisions.

$230k - $262k

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