Deep Learning Jobs

At Thinking Machines, we build tools that empower individuals to customize AI models for their specific needs, including the ability to train model weights. In this role, you will focus on frontier customization techniques and contribute to developing a leading post-training engine called Tinker. Your work will involve a comprehensive understanding of Reinforcement Learning (RL) science, and your findings will directly influence Tinker's training defaults, API design, and the open-source Tinker Cookbook. You will collaborate with internal research teams and contribute to open science initiatives for external partners.

$350k - $475k

Thinking Machines is building AI that extends human will and judgment, with a focus on training frontier models and developing tools that allow users to customize AI for their unique needs. This role leads the Fine-tuning Science team, focusing on frontier customization techniques and the Tinker post-training engine. It's a player-coach position where you will remain hands-on with scientific research while also guiding the team, shaping its direction, and ensuring research findings are integrated into Tinker. You will collaborate with internal research teams, contribute to open science, and engage with external users.

$350k - $475k

NVIDIA is seeking a Principal Systems Software Engineer for Semiconductor Inspection to join their team in Santa Clara. This role involves defining, building, and evaluating AI systems for semiconductor manufacturing, focusing on inspection and metrology. The work spans computer vision, time-series modeling, multimodal AI, and anomaly detection, with a strong emphasis on adapting and deploying these technologies within partner environments. The goal is to create integrated, reliable, and fast AI solutions that can be operated by fabs and equipment vendors, even in air-gapped facilities with limited data and tight production budgets.

$272k - $431k

Ramp is seeking an Applied Scientist Intern to join their team and contribute to solving critical business problems through machine learning. As an intern, you will be a fully integrated member, owning a project from inception to completion. You will collaborate with engineers, product managers, and business stakeholders to translate complex business needs into scalable, ML-driven solutions. This role offers the opportunity to apply ML in a practical setting, ship code, and create tangible value for Ramp and its customers, focusing on areas like credit, fraud, growth, or the core product.

NVIDIA is at the forefront of the AI revolution, exploring the next frontier of Physical AI, which involves machines that can perceive, reason about, and act within the three-dimensional world. The NVIDIA Cosmos platform provides open foundation models for Physical AI, designed to interpret images, video, and text to create a structured understanding of physical scenes, including motion, object interactions, geometry, and physical context. These models serve as the reasoning layer for robots, autonomous vehicles, and smart infrastructure. The Cosmos Engineering team is responsible for developing the core capabilities behind these models and is seeking a Senior Deep Learning Engineer to lead the data engine and end-to-end training verification for 3D spatial reasoning and perception. This role involves ensuring models accurately learn and represent 3D spatial concepts, working closely with research scientists to translate hypotheses into measurable capabilities in shipped models, and focusing on the critical role of data and evaluations in achieving frontier model quality.

NVIDIA is seeking versatile software engineers for its XLA team to build high-performance, production-grade software at the core of next-generation AI systems. This role involves developing compiler optimization algorithms for deep learning workloads, optimizing inference and training performance for the JAX framework and the OpenXLA compiler on NVIDIA GPUs at scale. You will collaborate with deep learning framework and hardware architecture teams to accelerate future AI software.

$152k - $288k

We are seeking a motivated engineer to enhance the integration of APIs across our CUDA-X math library software stack into modern LLM and agentic AI workflows. Organizations worldwide leverage GPU-powered data centers for AI, data analytics, and scientific simulations, driving progress in fields like LLMs, computer vision, and autonomous vehicles. Our team develops the essential GPU-accelerated libraries and SDKs for these technologies. In this role, you will ensure that human engineers and AI agents can easily build high-performance applications by effectively utilizing CUDA-X libraries. We are looking for individuals with a unique combination of technical expertise, developer experience, and communication skills.

$152k - $288k

Genmo is a research lab focused on developing advanced video world models to understand, simulate, and interact with the physical world, aiming to advance physical intelligence and enable robots to learn and act. We are seeking an exceptional Software Engineer to join our research team in pushing the boundaries of visual generative AI. In this role, you will collaborate with experienced researchers to develop and enhance generative models, ensuring their safe and effective deployment. This is an opportunity to contribute to cutting-edge AI research in a fast-paced startup environment.

LanceDB is seeking an AI Research Engineer to collaborate with their research team on open-ended research projects. The role focuses on end-to-end training flows across various AI domains, demonstrating how LanceDB can accelerate research. This is a unique opportunity to pursue research interests as an engineer, significantly impacting product development and raising awareness of LanceDB's capabilities.

$150k - $255k

NVIDIA is seeking a scientist at the intersection of quantum device physics, quantum calibration, and machine learning to help accelerate the quantum ecosystem. This role will focus on designing and building real-time models that learn from device physics, calibration experiments, decoding, and system performance. You will develop physics-informed data synthesis pipelines, post-trainable model architectures, and practical benchmarks for the quantum community. Your research will translate qubit physics and the quantum control stack into performant AI systems for fault-tolerant quantum computing, spanning synthetic training data generation, surrogate modeling, and co-optimized calibration-decoding pipelines. You will collaborate with teams across Product, Engineering, and Applied Research to push the frontier of Accelerated Quantum Supercomputers.

$192k - $305k

NVIDIA's Architecture group is seeking a GPU System Performance Architect passionate about GPU-accelerated deep learning, data analytics, and high-performance computing. This role involves designing and developing the next generation of GPU-accelerated computing systems, offering a significant impact within a dynamic, technology-focused company. You will contribute to innovative processor and system architectures, analyze system performance trade-offs using analytical models and simulators, and understand hardware-software interactions in large-scale data center deployments. Collaboration across software, product, and business teams will be key to guiding the future of GPU-accelerated cloud computing.

$224k - $357k

We are seeking senior engineers to contribute to the CUDA driver, a fundamental component for accelerating general-purpose computation on GPUs. This role involves enhancing features and improving performance to maximize NVIDIA hardware potential across diverse workloads, including deep learning, scientific computation, gaming, and virtual reality. The position requires strong system software programming, a deep understanding of operating systems and kernel programming, hardware architecture, and excellent communication skills. You will collaborate with hardware and software engineers to design, develop, debug, and deploy critical aspects of NVIDIA hardware and mobile system-on-chip devices. The work spans the entire development lifecycle, from design feedback and pre-silicon simulation to silicon bring-up and production software deployment.

$152k - $288k

NVIDIA is seeking a motivated system software engineer with deep understanding of device drivers and strong C/C++ skills to work on the CUDA Driver. This core component of our platform accelerates general-purpose computation on the GPU. You will be an integral part of a team delivering features and improvements to realize the full potential of NVIDIA hardware across diverse workloads, including deep learning, scientific computation, data science, gaming, and virtual reality. You will leverage your design abilities, coding expertise, and creativity to craft elegant solutions to exciting problems and shape the future direction of CUDA.

$184k - $357k

NVIDIA is seeking a motivated system software engineer with deep understanding of device drivers and strong C/C++ skills to work on the CUDA Driver. This core platform component accelerates general-purpose computation on the GPU. You will be an integral part of a team delivering features and improvements to realize the full potential of NVIDIA hardware across diverse workloads, including deep learning, scientific computation, data science, self-driving cars, video games, and virtual reality. You will use your design abilities, coding expertise, and creativity to craft elegant solutions to exciting problems and shape the future direction of CUDA.

$272k - $431k

Ambient.ai is seeking an AI Infrastructure Software Engineer to design, build, and optimize the AI infrastructure powering their real-time intelligence platform. This role involves working on systems to run state-of-the-art deep learning models, including computer vision, large language models (LLMs), large vision models (LVMs), and multimodal AI systems, across vast amounts of video data in real time. You will focus on scaling infrastructure for inference, evaluation, and continuous model improvement, partnering closely with research scientists and product engineering teams to bring AI advancements into production.

$168k - $205k

NVIDIA is seeking a Senior Software Engineer to specialize in Deep Learning Inference. This role involves designing, building, and optimizing GPU-accelerated software for AI applications, focusing on high-performance deep learning frameworks like SGLang and vLLM for efficient large-scale model serving and inference. You will contribute to improving these platforms, enabling the smooth deployment of advanced language models and working with the deep learning community to implement and release the latest algorithms. The work will concentrate on enhancing the performance of state-of-the-art LLM and Generative AI models across various NVIDIA accelerators, utilizing open-source tools and kernels for optimization.

$184k - $357k

Abundant is building the NVIDIA of training data, recognizing the critical need for scaling data as AI models advance beyond supervised learning to reinforcement learning and learning from experience. The founding team brings extensive experience from leading AI organizations, with a proven track record in deploying deep learning models at scale, training state-of-the-art models, and managing large data pipelines. As a Member of Technical Staff, you will collaborate directly with the founding team to architect the future of model reasoning and intelligence, working at the intersection of research and execution to push the boundaries of AI capabilities.

$120k - $120k

Parallel Wireless is seeking experienced wireless algorithm experts to research and design advanced cellular communication algorithms for 5G and beyond products. This role will leverage neural networks and machine learning techniques to reimagine mobile networks with innovative, energy-efficient Open RAN solutions. The ideal candidate will contribute to the future of telecommunications by driving innovation through green and sustainable networks.

NVIDIA is seeking software engineers for the CUDA Tile team to work on a new tile-based programming model for their GPUs. This role involves designing and implementing compiler transformations, developing MLIR-based dialects and lowering passes, and optimizing the performance of tile-based kernels across various NVIDIA GPU architectures. The work includes defining public APIs, crafting compiler and optimization techniques, and general software engineering tasks. NVIDIA is a leading technology company transforming computer graphics, PC gaming, and accelerated computing, now focusing on AI to define the next era of computing.

$152k - $242k

NVIDIA AI for Media is seeking outstanding engineers to join its team, focusing on building and optimizing real-time AI solutions for media and entertainment workflows. This role will involve setting the technical direction for content authenticity AI, architecting model and pipeline strategies for detecting synthetic and manipulated media, and making critical build-vs-adapt decisions. The team works on core technologies to solve ambitious computer vision and deep learning problems, aiming to deliver studio-quality audio and high-resolution video enhancement and effects.

$272k - $431k