Machine Learning Engineer Jobs

NVIDIA is seeking software engineers for the CUDA Tile team to support a new tile-based programming model for NVIDIA LPUs. This role involves designing and implementing compiler transformations, developing MLIR-based dialects and lowering passes, and optimizing the performance of tile-based kernels for various NVIDIA GPU and LPU architectures. The scope includes defining public APIs, crafting optimization techniques, and general software engineering. NVIDIA offers competitive salaries, equity, and a comprehensive benefits package, making it a desirable employer in the technology industry.

$152k - $242k

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

NVIDIA Research is seeking leading researchers in the area of simulation and AI for quantum. You should have a strong background of research excellence, software systems building experience, and interest in advancing the state-of-the-art in quantum computing and applications of GPUs to the quantum computing ecosystem. Specific areas of research interest include but are not limited to bespoke quantum representations, tensor network methods, quantum surrogate modeling, quantum inverse design, and automated proving for quantum systems. This role involves researching and developing creative and innovative algorithms and software systems for the quantum computing ecosystem, collaborating with research and product teams, and publishing original research.

$0k - $0k

NVIDIA Research is seeking PhD-level students for a research internship focused on AI hardware/software co-design, hardware accelerator architecture, IC design methodology, and VLSI design. The ideal candidate will possess a broad understanding of machine learning fundamentals, quantization and numerical methods for model optimization, digital VLSI circuits, and advanced VLSI design and verification methodologies, including the application of agentic AI to hardware design. This role involves contributing to cutting-edge research in AI accelerators and VLSI design, exploring innovative numerical methods, and potentially collaborating on research prototype testchips.

$0k - $0k

NVIDIA is seeking strategic, ambitious, hard-working, and creative individuals for its 2027 Deep Learning Computer Architecture internships. This 12-week full-time program offers an excellent opportunity to expand your career and gain hands-on experience with industry-leading Deep Learning Computer Architecture teams. Interns will work on projects with a measurable impact on NVIDIA's business, tackling complex challenges in AI and digital twins that are transforming major industries.

$0k - $0k

NVIDIA is seeking strategic, ambitious, hard-working, and creative individuals for its 2027 Deep Learning internship program. This 12-week full-time internship offers an excellent opportunity to expand your career and gain hands-on experience with industry-leading Deep Learning teams. Interns will work on projects with a measurable impact on NVIDIA's business, contributing to advancements in AI and digital twins that are transforming major industries. We are looking for students pursuing a B.S., M.S., or Ph.D. degree in a relevant field who are passionate about tackling complex challenges.

$0k - $0k

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

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

Stripe

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.

Salary not listed