PyTorch Jobs

Our team is building the machine learning backbone of the Perception component for NVIDIA DRIVE AV, focusing on solving complex problems for self-driving cars. We are seeking experienced Machine Learning Engineers with expertise in computer vision, LiDAR & camera perception, multi-sensor fusion, and AI infrastructure. The role involves inventing human-level AI for perception under real-world conditions, contributing to a defining technological advancement of our generation.

$184k - $357k

Baseten powers mission-critical inference for leading AI companies, enabling them to bring cutting-edge models into production. We are seeking inference performance engineers to make demanding AI workloads run faster and more efficiently. You will work across the entire stack, from the inference engine and runtime through scheduling, serving, and routing, applying advanced techniques to optimize performance. This role is ideal for someone who thrives in a fast-paced startup environment and is eager to make significant contributions to LLM inference.

$180k - $360k

We are a dedicated research lab focused on building, understanding, using, and risk-managing foundation models. Our mission is to advance AI research, cultivate the next generation of AI builders, and make significant contributions to a knowledge-driven economy. As part of our team, you will work on the core of cutting-edge foundation model training alongside world-class researchers, data scientists, and engineers. You will tackle fundamental and impactful challenges in AI development, contributing to groundbreaking AI solutions that have the potential to reshape industries. Strategic and innovative problem-solving skills are essential for establishing our institution as a global hub for high-performance computing in deep learning.

$150k - $450k

We are a dedicated research lab focused on building, understanding, using, and risk-managing foundation models. Our mission is to advance research, nurture the next generation of AI builders, and drive transformative contributions to a knowledge-driven economy. As part of our team, you will have the opportunity to work on the core of cutting-edge foundation model training alongside world-class researchers, data scientists, and engineers. You will tackle fundamental and impactful challenges in AI development, participating in the creation of groundbreaking AI solutions that have the potential to reshape entire industries. Strategic and innovative problem-solving skills are essential for establishing our institution as a global hub for high-performance computing in deep learning.

$150k - $450k

We are a dedicated research lab focused on building, understanding, using, and risk-managing foundation models. Our mission is to advance research, nurture the next generation of AI builders, and drive transformative contributions to a knowledge-driven economy. As part of our team, you will work on the core of cutting-edge foundation model training alongside world-class researchers, data scientists, and engineers. You will tackle fundamental and impactful challenges in AI development, contributing to groundbreaking AI solutions that have the potential to reshape entire industries. Strategic and innovative problem-solving skills are essential for establishing our institution as a global hub for high-performance computing in deep learning.

$150k - $450k

NVIDIA is seeking a Senior Applied Power Architect to join our Applied Power Architecture team. You will be instrumental in developing state-of-the-art GPUs for AI, Automotive, GeForce, and Mobile products. This role involves inventing methodologies, features, and techniques to tackle electrical, speed, and power challenges for next-generation AI architectures. You will drive power management initiatives to optimize efficiency and performance while adhering to electrical, thermal, and power constraints. The position requires a deep understanding of process characteristics and their impact on voltage-frequency (V-F) tradeoffs, as well as investigating multi-die technologies and proposing innovative solutions for packaging, interconnect, physical design, electrical, and thermal challenges. Collaboration with cross-functional teams, including ASIC designers and packaging experts, is essential for formulating effective solutions.

$168k - $311k

At Mach9, ML infrastructure engineers build and maintain the systems that power production AI models for civil engineering and surveying. Our ML pipeline spans 10,000+ miles of labeled survey data, image segmentation networks, and 3D prediction models serving real-time inference to surveyors and engineers in the field. This role is ideal for mid-career ML infrastructure engineers with experience building for both training and inference. You'll build training pipelines that handle deep transformer models on hundreds of terabytes of 3D point cloud and image data. You'll also architect our inference infrastructure, delivering both heavy offline detection algorithms and real-time responsive inference that integrates directly with our CAD software.

$160k - $250k

NVIDIA is seeking a Senior Perception Engineer to design and productize the next generation of autonomous driving perception systems. This role focuses on the core 3D obstacle perception pipeline, involving architecture and algorithm design, with a hands-on approach to implementation using modern transformer-based, multi-modal, and vision-language techniques. You will contribute to the development of state-of-the-art deep learning models for obstacle detection and tracking, leveraging multi-camera and multi-sensor fusion. The position requires defining objectives, prototyping architectures, running experiments, and adhering to best practices for training and evaluation, including techniques like large-scale pretraining and parameter-efficient fine-tuning. You will also help define and maintain performance metrics, analyze large datasets to identify failure modes, and contribute to the data strategy for perception, including model-assisted workflows and model-in-the-loop tooling. Collaboration with safety, systems, and software teams is essential to ensure solutions meet product requirements for safety, latency, resource usage, and robustness for large-scale deployment.

$184k - $357k

NVIDIA Research is seeking a world-class generative AI researcher to join its fundamental generative AI research team. The team focuses on advancing generative AI methods for scientific applications, including biomolecular design (small molecule, protein, and RNA design). The role involves fundamental research with the freedom to conduct ground-breaking work, aiming to push the boundaries of generative models and create real-world impact. Researchers will collaborate with a talented team that consistently publishes at top venues in machine learning and AI for science.

$168k - $265k

NVIDIA is seeking a Principal Perception Engineer to lead the design and productization of next-generation autonomous driving perception stacks. This senior individual contributor role involves setting technical direction for 3D obstacle perception, driving cross-functional execution, and mentoring engineers. The role requires deep hands-on involvement in architecture, algorithms, and implementation, with a focus on modern transformer-based, multi-modal, and vision-language techniques. The goal is to build intelligent machines that can perceive, reason, and solve complex problems using GPU-accelerated deep learning.

$272k - $431k

NVIDIA is seeking Senior Deep Learning Scientists to advance our work in streaming and agentic multimodal AI. This role involves developing models capable of reasoning, planning, and acting across diverse modalities, contributing to core algorithmic improvements for multimodal foundation models. You will work on high-impact, large language models and multimodal AI products that enhance user experiences, joining our Nemotron LLM team to tackle real-world agentic AI challenges.

$184k - $288k

NVIDIA is seeking Senior Deep Learning Scientists to advance our work in streaming and agentic multimodal AI. This role involves developing models capable of reasoning, planning, and acting across diverse modalities, with a focus on defining core algorithmic improvements for multimodal foundation models. You will contribute to high-impact, large language models and multimodal AI products that enhance user experiences, working within our Nemotron LLM team.

$152k - $288k

NVIDIA is seeking a Senior Radar Perception Engineer to contribute to the design and productization of their next-generation autonomous driving perception stack. This role involves working on the core 3D radar and multi-modal obstacle perception pipeline, focusing on architecture and algorithm design. The engineer will be hands-on with implementation, including modern transformer-based, radar-centric foundation models, and multi-sensor fusion techniques.

$224k - $357k

NVIDIA DGX Cloud provides AI services and endpoints for research and production workloads. We are seeking a Senior Production Engineer to develop software and automation that ensures these services are reliable, scalable, and safe to operate. The Production Engineering team focuses on large-scale distributed systems, including internal and external model endpoints, regional control plane services for workload orchestration and request routing, and the GPU/CPU compute infrastructure for inference and agentic workloads. Our work encompasses Kubernetes clusters across various cloud environments (AWS, Azure, Google Cloud, partner clouds) and on-premises deployments.

$184k - $357k

At Mach9, ML Engineers build the perception models at the core of our AI-enabled CAD system, extracting 3D object and line features from dense LiDAR point clouds and imagery. This product-focused role involves working closely with customers to develop Computer Vision/Machine Learning pipelines and workflows that solve their problems. You will be responsible for shipping new features and product lines end-to-end, from data strategy and model training to evaluation. This position is ideal for early-career engineers who learn quickly, are curious, and find satisfaction in shipping impactful products.

$180k - $300k

Exa is an applied AI lab building a novel search engine. We focus on massive-scale infrastructure for web crawling, training state-of-the-art embedding models, and designing high-performance vector databases. Our goal is to create perfect search over all the world's information, surpassing current capabilities. This role is critical to our search architecture, with flexibility in project assignments based on individual skills and interests. We are seeking a research engineer to tackle the complex challenges of understanding web page content, including parsing, classification, credibility assessment, and misinformation detection, all at a web scale.

$180k - $350k

Anthropic

Reinforcement learning (RL) is crucial for Claude's ability to reason, write code, and act autonomously over long horizons. This role is part of the team responsible for building and maintaining the RL training system, which is used to train production models and for researchers across Anthropic to run their experiments. The team collaborates closely with research teams company-wide on the science and engineering required to scale RL effectively. As a Research Engineer, you will be at the heart of RL at Anthropic, with a direct view of RL training at the frontier, as the system you own supports both production and research runs. You will work with collaborators across research teams to understand system performance, identify areas of fragility, and drive future improvements.

$500k - $850k

Ema builds AI Employees for HR, IT, and Finance, automating busy work across the employee experience from recruiting and onboarding to IT support and payroll. Our team, comprised of engineers from top tech companies and graduates from prestigious universities, is backed by leading investors and operates at the forefront of Agentic AI in production. We are developing systems where AI agents learn and improve from their real-world tasks, creating a continuous loop of production data, training, evaluation, and enhanced agent performance. The core challenge lies in determining the most effective interventions to improve agent behavior in live workflows.

SambaNova is seeking a Compiler Architect to join their team and drive innovation in compiler infrastructure and optimization algorithms. This critical role involves working with various layers of the compiler stack and coordinating with development teams to ensure state-of-the-art ML model performance on the SambaNova platform. Responsibilities include analyzing PyTorch and machine learning models to map operations onto the company's unique dataflow hardware.

$210k - $280k

SambaNova is seeking a talented and driven ML performance engineer to optimize and scale state-of-the-art foundation models on SambaNova's reconfigurable dataflow platform. You will work hands-on with advanced models to push the limits of throughput, latency, and efficiency. This role bridges the gap between deep learning and systems performance, collaborating across compiler, runtime, and hardware layers to deliver world-record performance for large-scale AI inference.

$180k - $255k