AI Engineering Jobs in San Francisco

NVIDIA is seeking a senior or principal engineer to join the Generalist Embodied Agent Research (GEAR) group, focusing on physics simulation for Project GR00T, an initiative to build foundation models and full-stack technology for humanoid robots. You will collaborate with a research team known for influential work in multimodal foundation models, large-scale robot learning, embodied AI, and physical simulation, contributing to impactful research projects and product roadmaps.

$224k - $357k

CoreWeave

CoreWeave is seeking a Senior Engineer to join its Benchmarking & Performance team. This role will focus on building performance insights and observability systems for AI infrastructure, covering everything from individual GPUs and interconnects to distributed training and inference workloads. You will be responsible for detecting, diagnosing, and surfacing performance signals across global data centers, transforming raw telemetry into actionable insights for engineers, product teams, and executives. This is a hands-on role focused on building observability tooling that provides near real-time answers about infrastructure performance, rather than a traditional data engineering or BI position.

$182k - $242k

NVIDIA is seeking a senior or principal engineer to build cutting-edge infrastructure for large-scale foundation model training within the Generalist Embodied Agent Research (GEAR) group. This team is spearheading Project GR00T, an initiative focused on developing foundation models and full-stack technology for humanoid robots. You will collaborate with a research team known for influential work in multimodal foundation models, large-scale robot learning, embodied AI, and physics simulation, contributing to impactful research projects and product roadmaps.

$224k - $357k

CoreWeave

As a Senior Forward Deployed Engineer (FDE), you will work directly with customers, embedding within their AI teams to help them build and deploy successful AI agents using the W&B Weave developer toolkit. This is a hands-on, customer-facing engineering role for individuals who thrive in high-impact environments, enjoy solving open-ended technical problems, and want to shape how leading organizations build with AI.

$182k - $242k

Join NVIDIA's global Developer Technology (DevTech) team as an AI Developer Technology Engineering Manager. In this role, you will lead a team of highly skilled engineers focused on accelerating the end-to-end performance of real-world Deep Learning and Machine Learning applications. You will also develop novel algorithms to optimally leverage NVIDIA hardware, working on applications such as Large Language Models (LLMs), Computer Vision, Speech, Recommender Systems, and Multimodal architectures. As a leader, you will define and drive strategic technical initiatives, setting goals and collaborating closely with customers, NVIDIA Research, and hardware/software teams to advance the state-of-the-art in accelerated AI.

$224k - $357k

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 is seeking Senior Software Engineers to join its Infrastructure, Planning and Process Team (IPP) to accelerate AI adoption across various engineering workflows. In this role, you will design and implement tools and software solutions that leverage Large Language Models and agentic AI to automate end-to-end software engineering workflows and enhance engineer productivity. You will be responsible for designing and implementing AI-driven optimizations within software development workflows to improve developer productivity, accelerate feedback loops, and enhance release reliability. Additionally, you will create and deploy predictive models to identify high-risk commits and forecast potential build failures.

$184k - $357k

We are seeking software engineers to work on next-generation graphics and computing products. Our charter is to develop the most demanding applications a GPU or high-performance computing server will encounter in its lifecycle, by collaborating closely with customers to support innovation in their factories and data centers. As a system software engineer, you will extensively use your knowledge of operating systems, algorithms, and computer architecture to provide robust and efficient solutions to validate and test next generation processors.

$184k - $357k

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 software engineers for its Deep Learning Frameworks Sustaining Engineering team. This role involves working directly with deep learning framework developers to integrate with open source codebases, contributing to software that powers breakthroughs in areas like image classification, speech recognition, and natural language processing. The team's work enables advancements used globally, making this a significant opportunity to impact the field of deep learning.

$152k - $288k

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

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

NVIDIA is seeking top research engineers for its Autonomous Vehicles Research team. This role focuses on AI models for autonomous driving, including agent behavior, end-to-end architectures, AI safety, closed-loop training, and AV foundation models. You will conduct fundamental, publishable research while also impacting products and collaborating with teams across various AI applications. The position offers a collaborative and nurturing research environment.

$184k - $357k

NVIDIA is seeking a world-class researcher in generative AI to join our research team. You will conduct original research for generative AI applications, including image, video, 3D, and audio generation. The role involves working with a team of researchers to make significant impacts with generative AI models, building research prototypes, and scaling them with large datasets and compute. After demonstrating the promise of your research, you will collaborate with product teams to integrate your ideas into products.

$192k - $357k

NVIDIA's GPU Architecture group is seeking world-class architects and software developers to lead architecture efforts. The role focuses on innovating in graphics and parallel computing to deliver the highest performance for deep learning and parallel processing algorithms. You will be responsible for developing and enhancing GPU architecture features, advancing parallel programming models, and designing new architectures to maintain leadership in AI computing. This involves interacting with other architects and researchers to build simulators, map deep learning workloads to hardware, and validate new architectural features.

$184k - $357k

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

NVIDIA Dynamo is a high-throughput, low-latency inference framework for serving generative AI and reasoning models across multi-node distributed environments. Built in Rust for performance and Python for extensibility, Dynamo orchestrates GPU shards, routes requests, and manages shared KV cache across heterogeneous clusters so that many accelerators feel like a single system at datacenter scale. As large language models rapidly outgrow the memory and compute budget of any single GPU, this platform enables efficient, resilient deployment of cutting-edge LLM workloads. We are seeking a Principal Systems Engineer to define the vision and roadmap for memory management of large-scale LLM and storage systems.

$272k - $431k

NVIDIA is seeking world-class deep learning researchers to build the next generation of AI for graphics and gaming. This role focuses on step-change research that delivers undeniable user experience improvements, driving fundamental AI research into products used by millions. You will conduct cutting-edge research, train advanced models, prototype in real-time engines, and collaborate with product, driver, and hardware teams to bring innovative ideas to market. If you are passionate about deep learning, real-time graphics, and delivering transformative technology directly into games and graphics applications, this is an excellent opportunity.

$168k - $265k

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), with the goal of creating research that pushes boundaries and has a real-world impact. This role offers the freedom and bandwidth to conduct groundbreaking research within a collaborative team that consistently publishes at top venues in machine learning and AI for science.

$192k - $305k

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

We are seeking a Senior System Software Engineer to contribute to the development of the Dynamo-Triton Inference Server. This role is part of NVIDIA's GPU-accelerated deep learning software team, which is building a high-performance AI inference platform to simplify the design and deployment of AI models. The platform aims to make AI models more accessible to all users and is crucial for enabling breakthroughs in various AI fields, including image classification, speech recognition, and natural language processing.

$152k - $288k

NVIDIA's Managed AI Research Superclusters (MARS) team is seeking a Senior GPU Supercomputer Scheduler Engineer to design and implement groundbreaking GPU compute clusters. This role involves developing solutions for demanding deep learning, high-performance computing, and computationally intensive workloads. You will contribute to the design and implementation of new scheduling features and services to optimize GPU compute clusters for fairness, occupancy, resilience, performance, and power usage. This is an excellent opportunity to deliver production-grade solutions, work with cutting-edge technology, and collaborate with technical leaders on significant challenges in machine learning, cloud computing, and system co-design.

$152k - $288k