AI Engineering Jobs in San Francisco

Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is within the Search & Indexing team, part of the Compute and Data Platform (AI Platform), responsible for making Wayve's multimodal driving data discoverable. The team builds indexing infrastructure for vector and metadata-based retrieval, powering data curation, training set construction, and ML research. You will be the first hire in this team, based in our Sunnyvale office.

$312k - $351k

Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is within the AI Validation team, a fast-growing group responsible for proving the safety and effectiveness of the Wayve AI Driver. The team focuses on simulation-led validation across various driving competencies. You will own the validation process for specific driving behaviors, developing metrics, automated tests, and analyzing results to ensure the AI Driver meets product requirements and safety standards. This role offers the opportunity to influence release decisions and contribute to bringing advanced autonomous driving technology to a global market.

$210k - $311k

OpenAI

OpenAI Presence is building a platform for reliable AI agents designed to handle high-volume business workflows in areas like customer support, sales, and insurance claims. The platform supports the entire agent development lifecycle, including evaluation, deployment, and improvement based on real interactions. It also integrates agents with company systems and supports voice and chat experiences, incorporating policies, guardrails, and human handoffs to ensure business control. We are seeking an engineer to help establish the foundational elements of Presence and develop systems that ensure AI agents are dependable in everyday business operations. This role involves setting the platform's technical direction while remaining hands-on with coding, taking architectural concepts from initial decisions through to scalable, reliable systems.

$230k - $385k

Wayve is building the AI platform for autonomous driving, enabling vehicles to learn directly from real-world experience. This role is on the Measurement team within AI Evaluation, focused on building and qualifying offline scene-understanding models that measure driving performance. These models leverage greater compute, larger models, and temporal context to create credible evidence at fleet scale, contributing to the development of a generalizable AI Driver for real vehicles.

$312k - $389k

Amplitude

The Analytics Compute Platform team at Amplitude owns the critical compute layer that powers every chart, insight, and experiment result seen by customers. This team sits between Amplitude's front-end analytics products and its proprietary analytics database, Nova, and is responsible for the compute API that translates user queries into fast, accurate answers. The core mandate is to enable a highly performant, reliable, and flexible system for computing insights, navigating the inherent tradeoffs between these goals. Day-to-day, you will collaborate closely with engineers across various product teams, including data management, marketing analytics, Session Replay, Experiment, CDP, and Query, as they are all consumers of the team's work.

$167k - $230k

Cerebras

Cerebras Systems is seeking a senior software engineer to build and scale the platforms and tools used to monitor and manage its growing fleet of AI clusters across multiple data centers. This role involves owning substantial engineering work across backend services, operational tools, and user-facing applications. You will collaborate with various teams, including Cluster Operations, infrastructure, inference, security, and data center teams, to translate operational challenges into reliable, scalable software solutions. The goal is to ensure fleet health, efficient issue response, and continuous compute capacity availability.

Salary not listed

Databricks

Databricks is seeking a Senior Software Engineer for its Files team to innovate at the intersection of storage, performance, and distributed systems. This role will directly shape the company's filesystem strategy to power next-generation agentic workloads. The Filesystem has emerged as a preferred runtime interface for AI agents, serving as a foundational layer for context, memory management, artifact authoring, version control, sandboxing, and inter-system communication, thereby accelerating AI adoption.

$166k - $225k

Anthropic

Our team studies how large teams of agents scale to solve complex problems that a single agent cannot, such as rewriting major codebases or formalizing mathematics. We investigate the impact of increasing agent numbers, compute budget, and task length on performance, cost, and coordination. This role bridges research and engineering, involving designing and running large experiments, building the necessary systems, and analyzing surprising results. We aim to move quickly from vague questions to running experiments, developing the platform and evaluations Anthropic uses to measure and manage large agent teams.

$500k - $850k

Together AI

Together AI is seeking a Staff Platform Infrastructure Engineer to lead service and platform infrastructure strategy across the engineering organization. This is a hands-on technical leadership role with the opportunity to establish a new team and define the standard for how services are built and operated. You will provide a stable, scalable service infrastructure foundation for mission-critical product and platform teams, focusing on reliability, repeatability, and ease of adoption. You will own and evolve crucial shared infrastructure, roll out new patterns and platform capabilities, and ensure services are built on company-wide foundations.

$240k - $280k

HeyGen is seeking an Applied AI Engineer to join the Video Agent team. This role focuses on enhancing the capabilities of an AI agent that transforms prompts, documents, or assets into finished videos. You will be responsible for improving the agent's ability to understand user requests, redesigning tools it utilizes, building evaluation frameworks, and shipping new features across the full stack. This position offers a unique opportunity to work directly with in-house trained models and researchers, owning features from conception to launch and making critical decisions about where to apply AI versus deterministic code.

$180k - $240k

Thinking Machines Lab

We are seeking a Software Engineer, Inference to be responsible for the reliability, scale, and efficiency of the systems that serve our AI models to users. This production-facing systems role is central to the company, acting as a bridge between cutting-edge inference techniques and the practicalities of serving live traffic, including rollouts, capacity management, and incident response. You will ensure that performance gains from our research and inference teams reach production safely and remain operational, powering our live serving platform and related products.

$300k - $400k

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

The AI Cloud and Developer Productivity organization delivers and maintains the tools and services engineers use daily to enhance the entire development process, from code checkout to deployment. The AV Developer Tools team specifically owns AI-native tools and services that enable consistent and secure agentic workflows with visibility into their usage. The goal is to accelerate AV development by supporting developer workflows for performance and observability, ensuring production-level reliability and availability. This role focuses on building AI-native tools and services to improve developer productivity.

$180k - $284k

Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role focuses on advancing model-based autonomous driving, improving core driving performance, and developing intelligent features. You will lead critical initiatives to design and deliver ML-driven behaviors that scale from assisted to autonomous driving, working across model architecture, data pipelines, evaluation frameworks, and real-world deployment. You will collaborate closely with AI Platform, Simulation, Robot Software, and Model Release teams to build performant, adaptable systems ready for production, while helping shape the long-term technical direction across Autonomy.

$312k - $351k

As an intern, you will help develop and optimize our AI training platform by improving data processing pipelines, accelerating model training and inference workflows, and/or enhancing testing infrastructure. You will support tool development, instrumentation, and system monitoring to boost reliability, reduce latency, and increase iteration speed for advancing autonomous driving performance. This role is part of GM's AV/AI Platform organization, which builds the software foundation and engineering tools that power safe, reliable, and scalable autonomous driving and advanced driver-assistance systems.

$97k - $142k

Wayve is seeking a founding Machine Learning Software Engineer for its Multi-Embodiment Generalist Agent (MEGA) team. This role focuses on building foundation models for general-purpose robots, enabling them to learn from diverse data and generalize across tasks and embodiments. You will be instrumental in designing and constructing the software systems that support MEGA's machine learning research, including scalable ML pipelines, data and training infrastructure, and shared tools for rapid experimentation. This position offers a unique opportunity to work closely with large-scale video and robotics datasets, modern ML frameworks, and advanced models, ensuring the reliability and maintainability of research systems as the program expands. You will own critical components of the software and ML infrastructure, influencing the team's ML stack development and operations.

$312k - $420k

Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is for a founding member of the Multi-Embodiment Generalist Agent (MEGA) team, focused on building foundation models for general-purpose robots. These models will learn from large-scale video, language, and robot-interaction data to generalize across tasks and embodiments, enabling agents to perceive, reason about, and act reliably in the physical world. The work combines frontier ML research with direct real-world evaluation on a growing fleet of robots.

$311k - $420k

Wayve is seeking a founding Research Scientist to join the Multi-Embodiment Generalist Agent (MEGA) team. This role focuses on building foundation models for general-purpose robots beyond self-driving vehicles, creating intelligent agents capable of perceiving, reasoning, moving, and manipulating the physical world across diverse embodiments. You will play a senior, high-impact role in defining the research agenda, technical strategy, and foundations of a new robotics program. The work involves the entire robot-learning stack, from scalable data flywheels and new model architectures to training large models, designing evaluations, and deploying policies on real robots. You will collaborate with researchers, ML engineers, and roboticists to advance ambitious ideas from research hypotheses to large-scale experiments and real-world capabilities.

$370k - $513k

The Evaluation Tools group builds internal products that accelerate the AI Driver development loop, from defining tests to debugging model behavior. Model developers, researchers, and QA engineers rely on these tools to understand driving performance and scale evaluations across millions of scenarios. You will join the Search & Agents squad, developing search, scenario mining, and agentic tooling that allows users to find driving scenarios and create tests without writing SQL. This role involves working across the full stack, owning features end-to-end, from user interaction and idea shaping to building robust software and validating impact. You will have opportunities to define new projects as user needs evolve.

$210k - $311k

Wayve is building the AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is within the Performance Architecture team, which focuses on making Wayve's AI workloads faster and more efficient across training and cloud inference. The goal is to unlock new product capabilities by enabling the training of larger models faster and running inference more efficiently at scale. The team plays a crucial role in the company's mission to make autonomy universal and accelerate the transition to a safer, more intelligent future of mobility.

$370k - $395k

d-Matrix

d-Matrix is at the forefront of software and hardware innovation, aiming to unleash the potential of generative AI. We foster a culture of respect, collaboration, humility, and direct communication, valuing diverse perspectives to create better solutions. We are seeking passionate individuals driven by execution to join us in shaping the future of AI. As a Staff-level individual contributor on the Software Infrastructure team, you will drive the automation for our CI/CD and release pipeline. This involves diagnosing and fixing pipeline failures, such as identifying real regressions, flaky hardware, or infrastructure issues, and pinpointing the exact merge that caused overnight test failures. You will blend AI techniques with traditional software automation to build highly reliable systems and architect compute and storage solutions across cloud and on-prem environments. Your role will involve partnering with various teams to develop and spread best practices, reducing developer toil and accelerating development velocity.

$155k - $235k

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

Brex

Brex is seeking an AI Engineer to join their Finance Agents team and build the Agent Workbench. This role involves creating a conversational surface where finance professionals can interact with AI agents through various documents and tools to understand financial activity, investigate findings, and steer agent actions. The initial focus will be on evolving the workbench into an extensible platform supporting multiple agent experiences, owning the extension model, integration protocols, and application architecture. You will collaborate with cross-functional teams to build across application layers, establish an agentic design system, and iterate with customers on features like policy editing, reporting, and data exploration.

$216k - $240k

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