AI Engineering Jobs in San Francisco

Lead performance optimization for large-scale AI workloads, transforming them into faster and more efficient systems. This role involves analyzing and optimizing AI training and inference across distributed systems, connecting compute, memory, and networking behavior. You will lead complex performance workstreams, identify bottlenecks, validate improvements, and translate performance data into actionable engineering decisions. The position requires building benchmarks, profiling demanding workloads, and developing C++ and Python tools for analysis and optimization, ultimately helping teams enhance efficiency, scalability, and reliability. This role offers broad technical scope across AI infrastructure at data center scale.

Salary not listed

Salesforce is seeking a Lead User Researcher to join their Digital User Research team, focusing on AI and Digital Experiences. This role will own research for agentic experiences and event technology, aiming to understand user mental models, unmet needs, friction, and trust in AI agents. The researcher will also lead initiatives to improve digital experiences for Salesforce events. The ideal candidate is a curious and experienced researcher adept at combining qualitative and quantitative methods to answer complex questions, influence product strategy, and communicate insights effectively.

$173k - $260k

We are seeking an intern passionate about developing tooling and productivity to join our AV Developer Productivity Organization. This role focuses on enhancing the tools and services that engineers use daily, including building and leveraging AI/Agentic tools to improve developer productivity. The intern will partner with other engineers, take feedback, and deliver reliable solutions to tasks, contributing to improvements in AV development toolchains and services that directly impact engineering productivity and company metrics.

$115k - $142k

NVIDIA is seeking a Software Engineer, Performance Instrumentation to join their team in Santa Clara, CA. This role offers an exciting opportunity to contribute to groundbreaking projects at the intersection of AI and accelerated computing. You will be instrumental in defining and maintaining performance data requirements, identifying and addressing visibility gaps, and ensuring the quality and efficiency of instrumentation APIs. This position requires a strong background in software engineering with a focus on performance and data quality within machine learning or systems software environments.

$168k - $270k

Harvey is transforming legal and professional services by combining frontier agentic AI, an enterprise-grade platform, and deep domain expertise. This role is an opportunity to help build a generational company at an inflection point, with strong product-market fit and investor support. The team moves fast, takes ownership, and is deeply committed to the mission, operating with intensity and pushing for excellence. As a Member of Technical Staff, Infrastructure, you will build the systems that power how we adapt and improve leading open-weight models for legal work, focusing on scaling post-training efforts and owning the infrastructure behind distributed training, RL rollouts, agent environments, and internal tooling for data management, evals, and job tracking.

$230k - $340k

The Imagine team is redefining AI-driven media experiences for Grok users worldwide. As a Product Safety Engineer on the Imagine team, you will build the critical safety systems and infrastructure that ensure Grok’s multimodal generation capabilities (images, video, audio, and beyond) are powerful, delightful, and responsibly deployed. You’ll design and scale safeguards, detection systems, and evaluation frameworks that prevent harm while preserving creativity and user freedom. Your work will directly shape how millions of users experience generative media — balancing rapid innovation with rigorous safety, trustworthiness, and alignment with SpaceXSI's mission. If you’re a driven engineer passionate about AI safety, content integrity, and building robust systems that protect users without stifling innovation, this is your opportunity to make a lasting impact at the frontier of multimodal AI.

$180k - $440k

Customer Experience Engineering is a critical intersection of customers, product, and engineering. In this role, you will tackle complex technical challenges and manage high-stakes customer situations from initial investigation through to resolution and prevention. You will leverage your deep technical expertise along with strong judgment and communication skills to guide customers, especially when solutions are not immediately apparent. Working from a shared queue, you will take ownership of problems requiring in-depth investigation, solution design, or cross-team collaboration, while also contributing to the team's overall response to customer needs. A key aspect of this role involves translating learnings into practical solutions, tools, and AI-enabled workflows to minimize recurring issues.

$120k - $135k

Stripe

Stripe is seeking product-minded engineers to join the Builder Experience Team, which focuses on improving how Stripe's product teams build, ship, and iterate on products. This team is AI-native and reimagines workflows using AI agents and automation to address fragmented and manual processes. The role involves identifying high-friction areas in the product development lifecycle, prototyping new approaches, and scaling successful experiments into solutions used across Stripe. You will work across the full stack, combining traditional software engineering with emerging AI capabilities to discover the right product, interface, and technical architecture in ambiguous problem spaces.

Salary not listed

Harvey is transforming legal and professional services by combining advanced agentic AI, an enterprise platform, and deep domain expertise. This role offers a unique opportunity to contribute to a rapidly scaling company at a pivotal moment, defining a new category in real-time. The team operates with intensity, ownership, and a commitment to excellence, driven by values of decisiveness, simplicity, and continuous improvement. If you are motivated to do your best work alongside driven individuals, Harvey is the place to build the future of professional services.

$230k - $340k

SpaceXSI is building the high-performance inference platform that serves Grok to millions of users daily with exceptional speed and reliability. As a Member of Technical Staff - Inference, you will be instrumental in designing and optimizing large-scale model serving systems from the ground up. This includes owning distributed infrastructure components like global KV cache, continuous batching, load balancing, and auto-scaling, as well as delving into deep low-level optimizations such as GPU kernels, quantization, and speculative decoding. This is a high-impact role where your contributions will directly influence the speed and reliability of user interactions with Grok at a massive scale.

$180k - $440k

SpaceXSI is building Super Intelligence to help humanity understand the universe, and you will join the multimodal team to push toward superhuman multimodal intelligence. This role involves advancing understanding and generation across modalities—image, video, audio, and text—spanning the full stack from data curation to end-to-end product experiences. You will collaborate cross-functionally to deliver frontier capabilities in multimodal reasoning, world modeling, tool use, agentic behaviors, and interactive human-AI collaboration, contributing to models that can see, hear, reason about, and interact with the world in real time.

$180k - $440k

Glean is seeking Machine Learning Engineers to join the Enterprise Brain team, focusing on building proactive AI products that revolutionize enterprise workflows by automating tasks and enhancing user productivity. This role involves leveraging deep user understanding and state-of-the-art enterprise graphs, utilizing LLMs, advanced ML techniques, agent orchestration, and cutting-edge ranking methods. You will contribute to a secure, open platform that redefines how enterprises find, use, and act on knowledge, with a focus on enterprise trust and scalability.

$200k - $280k

Moveworks is seeking a Senior Staff ML Engineer to join our AI Safeguards team. In this role, you will be instrumental in designing, building, and shipping critical safeguards within our agent harness. Your work will focus on protecting against prompt injection, single and multi-turn jailbreaks, toxicity, sensitive topics, and sensitive data exposure. You will develop robust evaluation frameworks to measure the efficacy of these safeguards, addressing both false positives and false negatives using diverse datasets. Collaboration with our Security team will be key, translating their findings into actionable eval cases and fixes, and co-developing mitigations. You will also select, integrate, and fine-tune detection models within a shared stack serving both commercial and government environments, while carefully balancing quality, latency, and cost for controls operating in the request path. Additionally, you will build configurable safeguard policies to empower customers in managing sensitive topics and monitor safeguard behavior in production, addressing any regressions.

Salary not listed

Crusoe

Crusoe is seeking a highly skilled Software Engineer to join their Data Center Infrastructure Engineering team. This role focuses on developing software for managing a fleet of GPU servers and the data centers that house them. The position involves creating and implementing advanced diagnostic, observability, automation, and repair tooling for high-performance GPU compute clusters. The ideal candidate will be a hands-on problem solver comfortable working independently, playing a critical role in maintaining the health and scalability of Crusoe's rapidly growing GPU fleet.

$170k - $205k

MatX is building the compute platform for AGI, developing vertically integrated full-stack solutions from silicon to systems. We focus on hardware and software to train and run the largest ML workloads for AGI, primarily using the Rust programming language. This role involves designing and optimizing kernels that interface directly with our hardware, working in partnership with ML Research and Hardware Engineering teams, and providing expertise on hardware architecture from a programmer's perspective to ensure seamless integration with the software stack.

$160k - $275k

Figure is an AI robotics company developing autonomous general-purpose humanoid robots with the goal of shipping robots with human-level intelligence. These robots are engineered to perform a variety of tasks in home and commercial markets. We are seeking a Staff AI Inference & Acceleration Engineer to join our Platform Software team. This role will be responsible for the on-board inference architecture for our humanoid robots, acting as the technical authority on how AI workloads are mapped, optimized, and executed across the robot's compute hardware. The primary objective is to reduce power consumption and cost while meeting strict latency and reliability demands for a real-time autonomous system.

$180k - $275k

OpenAI

The Human Flourishing team is dedicated to ensuring AI systems understand and advance human flourishing, aligning AI capabilities with human values and long-term goals. This role involves measuring and improving models' understanding of human values, supporting user agency, and helping users achieve their long-term objectives. The team also focuses on scaling tangible impact in domains like education and mental health, working across research, product, and partnerships to benefit a large user base. Recent work includes Beneficial RL and the MentalHealthBench benchmark.

$380k - $500k

Athelas

Commure is building the AI Operating System for healthcare, transforming how care is delivered, documented, and financed. Our platform spans the full care journey with Ambient AI, intelligent Agents, and autonomous RCM processing. We are at the center of transforming healthcare's $1 trillion administrative burden, with over 500,000 clinicians trusting our platform. Our Air AI team specifically is developing the intelligence layer within the clinical workflow, enabling listening, documentation, reasoning, and action to alleviate clinician burden. This involves real-time transcription, a conversational clinical assistant, and an AI Skills Library for automating clinical tasks.

$170k - $230k

Anthropic

Our team builds the systems that enable every change to Anthropic's software to reach production safely and predictably. This includes managing pipelines, staged rollouts, health-gated promotion, rollback, runtime configuration, and feature flags across a large fleet of machines. You will own a significant part of this end-to-end process, from initial design to ongoing operation in production. If you are passionate about influencing the speed and safety of Anthropic's software releases, this role offers a unique opportunity.

$405k - $485k

AssemblyAI

AssemblyAI is seeking a Senior GTM AI Engineer to join their revenue organization. This role involves building the essential pipelines, internal tools, and AI agents that drive go-to-market strategies. You will be responsible for identifying target companies, personalizing outreach, monitoring product usage for expansion or churn signals, and maintaining CRM data integrity. The primary goal is to empower the sales team to operate with three times their current capacity by leveraging software and AI solutions.

$160k - $220k

The Voyage AI team at MongoDB is building state-of-the-art embedding models and rerankers for efficient unstructured data search and retrieval, crucial for RAG, recommendation systems, and semantic search. This team, comprised of AI researchers from top universities, was recently acquired by MongoDB and is integrating its advanced embedding models with MongoDB's data platform to create powerful end-to-end solutions. We are seeking a Senior Research Scientist to join this dynamic team and contribute to the development of next-generation AI models, tackling complex challenges at the intersection of machine learning research and the practical deployment of large neural networks.

$126k - $248k

Sift

As a Machine Learning Engineer at Sift, you will bridge the gap between data science and large-scale distributed systems. You will build end-to-end pipelines that extract signals, train custom models per merchant, and serve predictions at production scale with low latency. You will work on an automated machine learning ecosystem that dynamically recalibrates models based on streaming global telemetry data.

$170k - $240k

SpaceXSI is building one of the world's largest AI supercomputers from the ground up. As part of the Compute Infrastructure team, you will own both the raw GPU supercomputer and the platform layer that runs on top of it. You will work across the full stack—from low-level GPU kernel optimizations and Linux kernel internals to massive-scale orchestration and virtualization—to make training and inference at SpaceXSI as fast, reliable, and scalable as possible. This is a broad, high-impact role that combines hardcore supercompute and compute infrastructure work. Your contributions will directly accelerate Grok's training speed and overall AI progress.

$180k - $440k

Perplexity AI

Perplexity is seeking a Member of Technical Staff to join their GPU Cluster Infrastructure team. This role will own the underlying GPU infrastructure that powers Perplexity's AI inference requests, which serve hundreds of millions of queries monthly. The goal is to abstract away the complexity of GPU provisioning, cluster configuration, and multi-cloud management into a unified, self-serve platform for both training and inference workloads. You will be responsible for the entire lifecycle of the GPU fleet, ensuring reliability, efficient capacity utilization, and seamless operation across different cloud providers.

$250k - $485k