AI Engineering Jobs in San Francisco

Asana

Asana's AI organization is building the platform and products that integrate AI into how teams work. This includes AI Teammates, an orchestration layer for agents, and the AI platform supporting all Asana AI features, covering model access, retrieval, context, evaluation, and production quality. We are seeking a Senior Staff Software Engineer to lead technical direction across this organization, tackling complex cross-team challenges, making critical architectural decisions, and shaping the future of AI within Asana's products. This role involves influencing strategy across multiple engineering teams located in San Francisco, New York, Vancouver, and Warsaw.

$297k - $350k

CoreWeave

CoreWeave is seeking a Staff Software Engineer to join the Inference Platform Team. This team builds and operates CoreWeave’s Kubernetes-native inference platform, designed for low-latency, high-throughput AI workloads at massive scale. As a technical leader, you will drive architecture, performance, and reliability across multiple services and teams. Your responsibilities will include optimizing inference performance (latency, throughput, and GPU utilization), improving system reliability, and working deeply in distributed systems and Kubernetes-based infrastructure, focusing on areas like scheduling, batching, and memory optimization. This role requires hands-on technical leadership and the ability to influence engineering direction across the organization.

$188k - $275k

NVIDIA is seeking a Manager, Deep Learning Inference Software to lead a team focused on advancing AI model deployment. This role involves shaping software for sophisticated AI systems, including large language models and multimodal generative AI, accelerated on NVIDIA GPUs. The team develops and optimizes open-source frameworks like vLLM/SGLang and FlashInfer to make AI deployment scalable, efficient, and accessible, enabling real-time inference at all scales. The position offers a chance to work with forward-thinking individuals and provides opportunities for career advancement in a rapidly growing company.

$184k - $357k

SpaceX is building Super Intelligence to help humanity understand the universe, developing the full Super Intelligence stack from supercomputers to large language models and applications. This role focuses on the most critical post-training and reinforcement learning challenges, including reward modeling, preference optimization (RLHF/DPO), and RL for improving reasoning, truthfulness, and real-world capabilities. You will work on cutting-edge AI problems with the ultimate goal of making life multiplanetary and expanding consciousness to the stars.

$180k - $600k

Stripe

The Link team is building the premier consumer network for next-generation commerce, leveraging over 300 million registered consumers to pioneer the future of Identity, Consumer Finance, and AI-driven transactions. The Consumer Preference team specifically focuses on enhancing the Link experience across web applications and agents, creating a wallet product that empowers consumers with financial insights, rewards, and secure transaction capabilities. As a Staff Full Stack Engineer, you will lead the design and delivery of new consumer features, taking ambiguous product opportunities from initial concept through validation and final launch. This hands-on role involves owning large, cross-functional projects, leading engineering teams, and making critical architectural decisions to balance quality with rapid iteration.

Salary not listed

CoreWeave

CoreWeave is seeking a Software Engineer to join the Inference team, focusing on shipping production features that enhance model serving on their GPU platform. This role involves improving latency, reliability, and cost-efficiency. As an IC1, you will implement well-scoped changes, learn operational practices, and grow with mentorship from experienced engineers. CoreWeave is the essential cloud for AI, providing a platform of technology, tools, and teams that enable innovators to build and scale AI with confidence.

$92k - $135k

NVIDIA is seeking a Senior Developer Relations Manager to drive the adoption of its Cosmos 3 and Cosmos world foundation models within the Robotics ecosystem. This role involves building strong technical relationships with strategic partners and the NVIDIA Cosmos Coalition community, guiding developers in applying these models for robotics and autonomous systems. The ideal candidate will possess deep technical knowledge, adaptability, and expertise in guiding architecture and implementation plans for physical AI, world foundation models, and robot learning.

$184k - $357k

SpaceXSI is building Super Intelligence to help humanity understand the universe. Our teams are building the full Super Intelligence stack: from our Colossus supercomputers, to Grok models, to our suite of applications, including Grok Bot, Grok Build, Grok Imagine, Grok Voice, and X. We're creating a powerful flywheel where more compute trains better models, better models power more useful products, and what we learn from people using those products shapes what we build next.

$180k - $440k

CoreWeave

CoreWeave is seeking a curious, energetic, and innovative Software and AI Engineer to join their Strategy and Transformation team. This role focuses on building intelligent internal systems, AI-powered workflows, and first-party applications that enhance how CoreWeave operates. You will partner with various departments like Finance, Supply Chain, HR, and Legal to integrate enterprise platforms and develop AI solutions. This position is ideal for an early-career engineer looking to grow at the intersection of software engineering, product development, and applied AI, contributing across the full development lifecycle from implementation to production support.

$120k - $160k

Join NVIDIA's Analytics and Data Intelligence (ADI) group to develop "libcudf," an open-source CUDA C++ library that accelerates database and DataFrame operations. This role involves driving speed-of-light performance in structured data processing across various hardware, from single workstations to multi-node GPU supercomputers. You will build the computational core for DataFrame and database accelerators, creating highly optimized C++ and CUDA libraries that leverage GPU parallelism for operations like data loading, parsing, joins, and aggregations. This is an opportunity to contribute to an open-source software suite and make a significant impact.

$184k - $357k

SpaceXSI is building Super Intelligence to help humanity understand the universe, creating a powerful flywheel where more compute trains better models, better models power more useful products, and what we learn from people using those products shapes what we build next. You will own and elevate our mission-critical research platform that directly accelerates SpaceXSI's frontier work. The evals team creates and executes instruments for measuring model capabilities and behaviors, used to clean datasets, provide feedback for experiments, and track growth throughout training. We also diagnose why agents fail and how to make improvements, managing execution infrastructure, designing data sets and grading schemes. You will design, build, and ship delightful and bulletproof systems that engineers love to use every day, valuing strong attention to detail, first principles reasoning, and great taste.

$175k - $275k

CoreWeave

CoreWeave is seeking a Senior AI Platform Engineer to lead the design, implementation, and governance of AI tools for employees. This role involves owning the AI tools used daily across various platforms, ensuring their safe adoption through integrated identity, device, and tenant-level controls. The ideal candidate is passionate about AI's bleeding edge, enjoys rapid learning, and has a strong background in production software/SRE engineering, hands-on LLM/agent work, and enterprise IT controls. Success in the first year means enabling employees with frontier AI capabilities quickly and securely, establishing a robust AI operating framework, and building foundational controls for AI tool adoption.

$182k - $242k

NVIDIA is building a unified foundation model that understands relational databases and heterogeneous graphs, enabling a new era of enterprise AI. This role involves designing, building, and evaluating novel Transformer and graph neural network architectures that generalize across diverse data schemas. The work will power applications like recommendation systems, demand forecasting, fraud detection, and predictive maintenance. You will collaborate with researchers and engineers across the entire machine learning lifecycle, from architecture exploration and large-scale training to post-training optimization and high-performance inference, translating foundational research into production systems.

$184k - $357k

As an ML Infrastructure Engineer, you will play a pivotal role in building and optimizing the reliable, high-performance ML platform that powers recommendations on X. We're looking for exceptional engineers who are passionate about our mission and have a strong desire to make a meaningful impact.

$180k - $440k

CoreWeave

CoreWeave is seeking a Senior Engineer for its Benchmarking & Performance team. This role will be instrumental in managing a planet-scale performance data warehouse, handling the ingestion, storage, transformation, and analysis of performance events across global data centers. The engineer will also contribute to industry-leading performance benchmarking publications like MLPerf. The position involves owning designs, elevating engineering standards, and driving measurable improvements in latency, throughput, and reliability across multiple services. Collaboration with product, orchestration, and hardware teams will be key to evolving the Kubernetes-native platform and meeting stringent P99 SLAs at scale.

$182k - $242k

NVIDIA is seeking a visionary Product Architect with strong expertise in systems architecture to join our AI Infrastructure and Agentic AI team. You will design and shape the architectures that connect the world's most powerful AI clusters, transforming ideas into well-defined, functional products by balancing technical feasibility with user needs and market demands. This role is at the nexus of designing and overseeing cutting-edge AI solutions from conception through their entire lifecycle, focusing on efficiency and multiplying efforts.

$184k - $357k

SpaceXSI is building Super Intelligence to help humanity understand the universe, developing the full Super Intelligence stack from supercomputers to large language models and applications. This role is at the intersection of engineering and data operations, focusing on ensuring engineering teams receive high-quality data for training and evaluation. You will partner with model teams, design projects to capture meaningful signals, and drive data yield and evaluation lift, influencing data strategy through hands-on technical work.

$144k - $220k

CoreWeave

CoreWeave is seeking a Senior Software Engineer focused on Inference to join their team. In this role, you will be an area owner, responsible for leading designs, elevating engineering standards, and delivering significant improvements in latency, throughput, and reliability across multiple services. You will collaborate with product, orchestration, and hardware teams to enhance our Kubernetes-native inference platform and meet stringent P99 SLAs at scale. This is an opportunity to work on a cutting-edge AI cloud platform during a period of hyper-growth, contributing to innovative solutions for complex problems.

$152k - $204k

NVIDIA is seeking a Senior MLOps Engineer to join our DSX Enablement team, collaborating closely with strategic customers to implement and enhance groundbreaking AI workloads. We partner with the world's most innovative AI companies and open-source communities to address their most challenging technical problems. In this role, you will develop innovative solutions that advance AI infrastructure capabilities, advise infrastructure experts on the demands of ML workloads, and help practitioners diagnose and solve full-stack AI and ML system problems. You will work on a team with direct responsibility for the success of internal and external customers’ AI and ML initiatives, including LLM performance evaluation and supporting new hardware in open-source frameworks.

$184k - $357k

CoreWeave

CoreWeave is seeking a Senior Software Engineer I, Inference to join their AI Cloud team. This role involves owning specific areas of the inference platform, leading design efforts, and improving latency, throughput, and reliability. You will collaborate with product, orchestration, and hardware teams to enhance a Kubernetes-native inference platform and meet strict service level agreements at scale. The position requires a strong focus on engineering standards, incident management, and mentoring junior engineers.

$139k - $204k

NVIDIA is seeking an experienced HPC-AI Engineer to join the Networking Clusters Solutions Infrastructure team. This role focuses on building supercomputers and AI clusters using cutting-edge technologies. You will be instrumental in contributing to breakthroughs in artificial intelligence and GPU computing by providing insights on large-scale system design and tuning. You will collaborate with researchers, developers, and customers, working with the latest accelerated computing and deep learning platforms to craft improved workflows and develop innovative solutions. The position involves interacting with specialists in HPC, OS, GPU compute, and systems to architect, develop, and deploy large-scale performance platforms.

$176k - $334k

CoreWeave

CoreWeave is seeking an early member for its Applied Training team to build the research infrastructure that enables AI pioneers to train models with confidence. This role addresses the problem of AI labs spending valuable research time on cluster setup and operations instead of innovation. You will contribute to the roadmap and develop solutions for our Kubernetes-native research cluster platform or sandbox client for agentic training and evaluation, aiming to provide frontier lab-level infrastructure to all CoreWeave customers.

$182k - $242k

NVIDIA is building the next generation of AI-native autonomous driving architecture, integrating classical safety stacks, foundation models, and scalable AI systems into a unified production platform. This role is for individuals passionate about shaping the future of autonomous systems over the next decade. The work involves the deep integration of classical safety architectures and large-scale AI-driven systems, moving beyond purely classical robotics or end-to-end neural networks.

$224k - $431k

CoreWeave

CoreWeave is seeking a Senior Engineer for its Benchmarking & Performance team, focusing on GPU kernel authoring and optimization. In this role, you will be responsible for writing, profiling, and tuning GPU kernels that are critical for large-scale model serving, aiming to maximize throughput and minimize latency. You will also contribute to industry-leading performance benchmarking publications. This position requires an individual who can lead designs, improve engineering standards, and deliver significant enhancements to latency, throughput, and reliability across the inference stack. You will collaborate with product, orchestration, and hardware teams to translate kernel-level optimizations into system-wide improvements and meet stringent Service Level Agreements (SLAs) at scale.

$182k - $242k