Machine Learning Engineer Jobs

Anduril Industries

We are seeking a Staff ML Tech Lead Manager to apply the latest research to solve our toughest problems. This role will be responsible for owning the entire machine learning stack for the Counter Intrusion team. You will design and train multi-sensor object detection models for perception on edge compute devices that push the boundaries of what's possible with our sensors. You will develop learning algorithms to optimize the behavior of autonomous systems. Your work will directly impact Anduril's ability to deliver cutting-edge defense technology, with the opportunity to identify and develop novel ML applications across our product portfolio.

$220k - $330k

CoreWeave

CoreWeave is seeking a Principal Software Engineer to own the frontend architecture of the W&B app, which is used daily by ML researchers and engineers. The role focuses on ensuring app performance, stability, and reliability by addressing challenges in data composition and load times. This is a frontend-led position requiring deep architectural understanding and cross-organizational collaboration. You will be responsible for defining architectural boundaries, shaping designs across teams, and executing plans to improve the user experience and product features.

$227k - $303k

NVIDIA is seeking a Deep Learning Communication Architect to join its software architecture group. This role involves scaling Deep Neural Network (DNN) models and training/inference frameworks to systems with hundreds of thousands of nodes. You will be instrumental in optimizing communication performance by identifying and eliminating bottlenecks in data transfer and synchronization. The position requires designing efficient communication protocols tailored for deep learning workloads and collaborating with hardware and software teams to craft systems that leverage high-speed interconnects and communication libraries. You will also research and evaluate new communication technologies to enhance performance and scalability, and develop proofs-of-concept to validate new strategies.

$184k - $357k

CoreWeave

CoreWeave is seeking a Principal Engineer to lead its Benchmarking & Performance team. This role will be responsible for a global performance data warehouse, ingesting, storing, transforming, and analyzing performance events across all data centers. You will also play a key part in industry-leading performance benchmarking publications, working closely with NVIDIA and the open-source community to demonstrate CoreWeave's performance and reliability leadership in AI.

$206k - $333k

NVIDIA's Seattle Robotics Lab is seeking a Senior Robotics Research Scientist to contribute to fundamental and applied robotics research across perception, planning, control, and learning. This role involves developing and integrating algorithms and models for robotic manipulation and loco-manipulation, aiming to transform research paradigms and transfer innovations into NVIDIA's robotics and simulation products. The scientist will play a pivotal role in achieving ambitious goals within the next 2-3 years, requiring exceptional research and engineering skills, a proven track record, and a team-first mentality. Collaboration with academic and product teams, as well as mentoring junior researchers, will be key aspects of this position.

$192k - $357k

Anduril Industries

Anduril Industries is a defense technology company transforming military capabilities with advanced technology. We are seeking Software Engineers specializing in Computer Vision and Machine Learning Engineering to solve challenging perception problems for our UAVs. This role requires hands-on proficiency in C++ and/or Python within a Linux environment, leveraging cutting-edge research and deep expertise in areas like Object Detection and Object Tracking.

$191k - $253k

Join the GPU Foundations Developer Tools team at NVIDIA and contribute to the development of groundbreaking performance analysis tools for GPUs and SoCs. You will innovate as you develop performance analysis capabilities for NVIDIA hardware, drivers, and SDKs, supporting current and future generations of GPUs. Your work will empower inventors worldwide to maximize the potential of our GPU hardware in their applications, directly impacting the performance profiling tools based on next-generation hardware that you help define.

$152k - $288k

Anduril Industries

Anduril Industries is a defense technology company focused on transforming military capabilities with advanced technology. Our core Lattice platform integrates sensors to provide real-time situational awareness. As a Software Engineer on the ML Infrastructure team, you will develop the core systems and infrastructure that perception engineers use to field autonomous edge systems. This includes developing data management systems for perception and autonomy, model training/orchestration systems, and an internal perception platform for data querying, model training, and evaluation.

$167k - $222k

We are seeking software engineers to design and develop libraries and tools that simplify and accelerate computing for unstructured sparsity in Deep Learning (DL) and High-Performance Computing (HPC). Our team develops GPU-accelerated libraries and SDKs that are crucial for AI, data analytics, and scientific simulations. In this role, you will contribute to solutions involving sparse tensor computations, domain-specific language (DSL) specifications for sparse storage formats, and on-demand code generation. The ideal candidate will have experience in accelerated computing software development and a passion for advancing the state-of-the-art in various accelerated computing domains and DL frameworks like PyTorch.

$184k - $357k

Nvidia is seeking a Distinguished Engineer for Wireless Infrastructure to lead technology strategy for next-generation wireless infrastructure, focusing on AI-RAN and Agentic Core. The role involves applying AI/ML to 6G radio access networks (RAN), transforming the wireless core into an agentic AI-based packet core, and driving rapid prototyping of GPU-accelerated algorithms. You will collaborate with industry partners and customers to bring innovations to market and work with standards bodies like 3GPP and O-RAN to integrate AI-native approaches into future technologies. The position also includes driving development for new applications in Integrated Sensing and Communications (ISAC), Semantic Communications, and Physical AI at the Edge.

$320k - $489k

Anduril Industries

Anduril Industries is a defense technology company focused on transforming military capabilities with advanced technology. We are seeking Senior Software Engineers to join our Tactical Recon & Strike team, which develops aerial small drones and associated equipment. This role will focus on solving complex perception problems for our UAVs, leveraging cutting-edge research in computer vision and machine learning. You will work on bringing products from concept to operational systems, collaborating with specialist teams to address critical customer needs. If you are passionate about creating powerful robotics platforms and tackling challenging technical problems, we encourage you to apply.

$220k - $292k

NVIDIA is seeking architects to design hardware accelerator and processor architectures that enable state-of-the-art machine learning and data analytics algorithms and applications on next-generation mobile, embedded, and datacenter platforms. This role offers the opportunity to make a real impact in a dynamic, technology-focused company. As a member of the deep learning architecture team, you will contribute to features that advance the state of AI on next-generation GPUs and systems. This position requires staying current with the latest DL research and collaborating with diverse teams, including DL researchers, hardware architects, and software engineers.

$224k - $357k

Anduril Industries

Anduril Industries is a defense technology company focused on transforming military capabilities with advanced technology. We are seeking a Machine Learning Engineer to join the Air Dominance & Strike team, which designs, builds, and flies autonomous air vehicles. This role is crucial for accelerating our design cycles by applying the latest research in physics ML to critical bottlenecks. You will own the entire surrogate modeling stack, from architectures and training infrastructure to data pipelines and user tooling, directly impacting hardware decisions for our air vehicle programs.

$220k - $292k

NVIDIA Research is seeking exceptional systems researchers to contribute to the development of hardware, software, and infrastructure technology for ML systems of all scales. Advances in AI/ML heavily rely on the development of efficient, scalable, resilient, and trustworthy systems for training, fine-tuning, and serving ML models. AI/ML applications are also pushing the limits of both personal devices and warehouse-scale data centers. All layers of AI systems need to be co-designed and co-optimized to maximize performance and energy efficiency, improve scalability, and support emerging algorithms in this space. We are seeking candidates that have a proven track record of research excellence, a broad perspective on systems design across the full hardware/software/infrastructure stack, and depth in one of the layers such as AI/ML systems, operating systems, distributed systems, data management, cloud computing, and computer architecture.

$168k - $265k

Anduril Industries

Anduril Industries is a defense technology company focused on transforming military capabilities with advanced technology. The Maritime Digital Production (MDP) team builds and deploys the full technology stack for Anduril's shipbuilding factories, including data infrastructure, manufacturing execution systems, scheduling engines, and AI systems. This role is for a Machine Learning Engineer on the Applied Intelligence initiative, acting as a bridge between ML research and production operations. You will own the infrastructure that brings ML prototypes to production-grade systems, working at the intersection of data engineering, MLOps, and production operations. The role requires translating ML requirements into production infrastructure, coordinating across teams, and serving as an on-call owner for deployed ML systems, ensuring AI applications function reliably 24/7 in real manufacturing environments.

$220k - $292k

NVIDIA is seeking engineers to develop algorithms and optimizations for our LPX inference and compiler stack. You will work at the intersection of large-scale systems, compilers, and deep learning, crafting how neural network workloads map onto future NVIDIA platforms. This is your chance to be part of something outstandingly innovative!

$152k - $288k

Anduril Industries

Anduril Industries is seeking a Machine Learning Engineer to join the Counter Intrusion MSE team. This role will own the machine learning stack for the team, focusing on designing and training multi-sensor object detection models for edge compute devices and developing learning algorithms to optimize autonomous systems. You will be responsible for the end-to-end design, implementation, and performance of the ML stack, including data collection and training infrastructure. Your work will directly contribute to Anduril's mission of delivering cutting-edge defense technology and will offer opportunities to identify and develop novel ML applications across the product portfolio.

$220k - $292k

NVIDIA is seeking a Senior ASIC Engineer to design and implement a Memory Management Unit (MMU) for world-leading SoCs and GPUs. This role offers the chance to impact products across various domains, from consumer graphics to autonomous vehicles, and contribute to the forefront of artificial intelligence and deep learning innovation. You will join a global team dedicated to pushing technological boundaries and shaping the future of computing platforms. As part of the MMU team, you will develop a state-of-the-art memory management system, crucial for delivering the performance demanded by high-end applications and supercomputers.

$136k - $265k

Anduril Industries

Anduril Industries is a defense technology company focused on transforming military capabilities with advanced technology. We bring the innovation of 21st-century companies to the defense industry, redesigning how military systems are developed and sold. Our systems are powered by Lattice OS, an AI-driven operating system that provides real-time command and control. As the world faces strategic competition, Anduril is dedicated to delivering cutting-edge autonomy, AI, computer vision, sensor fusion, and networking technology to the military rapidly.

$167k - $222k

NVIDIA seeks a Senior DL Performance Architect to join a pioneering group focused on pushing AI Inference performance boundaries through ambitious hardware-software co-design. This role offers an outstanding opportunity to develop world-class performance strategies, guide future GPU architecture decisions, and lead AI innovation. If you are passionate about AI efficiency, have a proven record of modeling LLM performance and architecting AI systems, and enjoy optimizing every cycle, this role may be perfect for you.

$152k - $288k

NVIDIA's DGX Cloud team is seeking a Senior Full-Stack Software Engineer to join the AI Hub team. This role will accelerate AI research by ensuring NVIDIA's AI infrastructure is used efficiently, transparently, and at scale. The primary goal is to build a unified, self-service portal for AI researchers to manage, monitor, and optimize their use of Managed AI research Superclusters. You will lead the architecture and delivery of high-scale web products, own multi-team initiatives, and drive reliability and performance improvements.

$224k - $357k

NVIDIA is seeking a Senior Robotics Research Engineer to join its Seattle Robotics Lab, focusing on AI and robotics for drug discovery. This role involves developing fundamental robotics technology and building real-world systems to automate and bring autonomy to molecular discovery, manufacturing, testing, and validation. You will work on creating digital twins of robots and lab environments, simulating robot physics, developing perception pipelines, and translating experimental protocols into executable physical procedures. The position requires training robots for complex manipulation tasks and integrating these systems into biological experiment workflows, collaborating with research scientists and engineers across NVIDIA and Eli Lilly.

$184k - $288k

Join the CPU performance architecture team to drive the development of CPU technology for diverse applications including AI/deep learning, high-performance computing, cloud services, gaming, virtual reality, and autonomous vehicles. This role involves pushing performance boundaries for all CPU products by analyzing and optimizing workloads across various markets and identifying critical application behaviors. You will work with architects to improve future CPU and system designs and benchmark NVIDIA's CPU offerings against competitors.

$224k - $431k

We are seeking a Senior Performance Modeling Architect to define and enhance our next-generation CPU Cache Hierarchies and interconnects. This role offers a unique opportunity to develop scalable solutions that bridge the high-reliability, low-latency requirements of Automotive systems with the efficiency and density demands of Data Center environments. You will be instrumental in creating the core models that manage data flow across our silicon, ensuring our L3/System Caches and coherent fabrics meet ambitious performance targets.

$152k - $288k