AI Engineering Jobs in San Francisco

Figure is an AI robotics company developing autonomous general-purpose humanoid robots with the goal of shipping robots with human-level intelligence for home and commercial markets. We are seeking a backend engineer to build the services and infrastructure for Figure's consumer mobile application, encompassing areas like authentication, payments, and tooling. You will be responsible for end-to-end ownership of systems, from identity and billing to delivery pipelines, internal dashboards, and the cloud platform.

$200k - $400k

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 both home and commercial markets. We are seeking a Reinforcement Learning Engineer to focus on the development, training, deployment, and evaluation of advanced reinforcement learning algorithms specifically for the whole-body control of our humanoid robots.

$150k - $350k

Mercor is seeking a Research Engineer Intern to work at the intersection of engineering and applied AI research. This role will focus on owning benchmarking pipelines, evaluation systems, and failure analysis workflows that directly inform how we train and improve frontier language models. Your work will be crucial in defining how we measure tool use, agentic behavior, and real-world reasoning. You will design and run evaluations, build rubrics and scorers, and translate failure analysis into actionable improvements for post-training, RLVR, and data pipelines.

Salary not listed

Athelas

At Commure, we are building the AI Operating System for healthcare, transforming how care is delivered, documented, and financed. Our platform integrates ambient AI, intelligent agents, and autonomous RCM processing to address the significant administrative burden in healthcare. We are at the forefront of this transformation, with a rapidly growing user base and recent significant funding, reflecting strong market confidence. Our team works directly with clinicians, ensuring immediate impact and a fast-paced, ownership-driven environment. We are seeking individuals energized by complex challenges and a commitment to excellence to help build the future of healthcare.

$170k - $190k

SpaceXSI is building Super Intelligence to help humanity understand the universe. We are seeking exceptional Applied engineers to join a high-priority project that approximately 600 million monthly users use. This is an exciting opportunity for individuals with an engineer or scientist background to apply their skills to recommendation systems, ranking algorithms, search technologies, and many other systems. You’ll work at the intersection of advanced AI development and real-world impact, enhancing the ability to connect users with relevant content, accounts, and experiences.

$180k - $440k

Together AI

We are seeking a software engineer with a strong focus on reliability to manage the production operations of our Kubernetes-native control plane, which provisions and runs our GPU inference fleet. This platform team develops the API and controllers that enable the inference team to define their needs, such as clusters, model deployments, or capacity changes, and ensures these are reconciled. Your role will be to maintain this functionality in the real world by troubleshooting issues like InfiniBand problems, diagnosing serving faults, and enhancing our automated self-healing systems to permanently resolve recurring failures. The primary goal is to ensure the decoupling remains effective even during disruptions and to maintain fleet efficiency without introducing new instabilities. You will be responsible for inference serving SLOs and error budgets, leading incident response, and implementing lessons learned from postmortems into system changes. Additionally, you will address the reliability aspects of capacity efficiency, recognizing that misconfigured defragmentation or bin-packing can pose production risks. We value candidates with hands-on production reliability experience in large-scale distributed or GPU-serving systems, proficiency in troubleshooting InfiniBand/RDMA and Kubernetes controller-based platforms, and a software engineering approach to GPU infrastructure.

$250k - $320k

We are seeking an experienced software engineer to join our Agentic AI Harness & Quality team, focusing on expanding customer-facing agentic AI capabilities. You will leverage modern agentic AI tools, including frontier LLMs and infrastructure for tuning, evaluating, and serving AI agents in production. This is an eval-driven, data-centric role where you'll contribute to building error-free, inclusive, and privacy-preserving datasets. Your work will focus on enhancing AI performance in production, encompassing accuracy, quality, latency, reliability, and the overall user-facing system. We aim to build the world's highest-performing enterprise assistant platform, and your contributions will be crucial to this objective.

Salary not listed

We are seeking a Lead Machine Learning Engineer with a focus on MLOps to join our Recommendation Engine team. In this role, you will be instrumental in building and maintaining sophisticated pipelines for distributed model training on large compute clusters, as well as for batch and real-time model serving. You will also manage hyperparameter tuning at scale, model monitoring, and production validation within a well-managed, controlled environment. Our product, Personalization and Insights, delivers high-throughput, low-latency applications powered by state-of-the-art machine learning architectures deployed on AWS. These applications enhance personalized experiences across Chase Consumer & Community Banking channels, integrating traditional banking services with offerings in Travel, Merchant Offer Shopping, and Dining.

Salary not listed

Figma is seeking an AI Engineer for its Internal AI team, which focuses on making AI adoption easier and more impactful for all Figmates across various departments like Sales, Support, Finance, People, and Legal. This role is a key part of the Applied AI Builds pillar, responsible for identifying high-value AI opportunities, prototyping solutions, and taking them to production. You will work directly with teams to understand business workflows, build AI agents and automations, integrate them with enterprise systems, and ensure their safe and reliable operation. This position involves owning solutions through production, determining the appropriate use of AI versus traditional software, and shaping how agents can safely interact with enterprise systems. You will also collaborate with platform teams to build reusable AI capabilities and help define Figma's approach to Applied AI engineering.

$153k - $296k

Databricks

Databricks is seeking a PhD GenAI Research Scientist Intern to join their Mosaic AI organization. This role focuses on pushing the frontier of domain adaptation for Large Language Models (LLMs) and AI systems, enabling them to work effectively within custom enterprise domains. You will tackle open research problems related to scaling and automating evaluation, fine-tuning with synthetic data, retrieval augmentation, and efficient inference. The intern will collaborate with the research team on projects involving adapting, improving, and evaluating existing methods, designing novel domain adaptation techniques, and composing multiple methods for efficient post-training.

$0k - $0k

Baseten

Baseten is seeking a Software Engineer to join the Agent Infra team, focusing on building and enhancing Agent Drive, a distributed filesystem critical for agent sandboxes. This role involves developing a storage solution that supports thousands of concurrent sandbox mounts with strict data isolation and consistency. The work leverages advanced storage techniques like tiered caching, replication, and content-addressed data, with a focus on Linux filesystem internals. The team operates with a high degree of ownership, shipping features rapidly and contributing to the core infrastructure that powers leading AI companies.

$160k - $330k

NVIDIA is seeking a Senior Software Engineer to design and develop AIOps and Observability platforms. These platforms are crucial for monitoring, diagnosing, and optimizing millions of internal assets and services across various environments including cloud, on-premise, data centers, supply chain, and edge. You will collaborate with a team of engineers, product managers, and partners to shape the observability strategy, roadmap, and best practices for NVIDIA. Additionally, you will mentor other engineers in observability, machine learning, and related tools and techniques.

$200k - $322k

The AI Center of Excellence is seeking a Senior Principal AI Engineer to serve as a technical expert and thought leader for Mastercard’s enterprise AI infrastructure. This role will shape the architecture and engineering of secure, scalable, production-grade platforms supporting LLM model training and inferencing, machine learning models, large language models (LLMs), agentic AI, and emerging AI capabilities in production. The successful candidate will bring deep expertise in high-performance computing (HPC), GPU and accelerated computing, rack-scale systems architecture, private and hybrid cloud, Kubernetes, high-performance networking and storage, observability, security, and reliability. They will work closely with AI Engineers, Data Scientists, Platform Engineers, Security, Architecture, and Product partners to deliver resilient, cloud-native AI platforms in a highly regulated environment.

$254k - $407k

Capital One is seeking a Director AI Engineer to lead the creation of responsible and reliable AI systems that are transforming the banking industry. This role is central to the Intelligent Foundations and Experiences (IFX) team, which advances the state of the art in AI science and engineering, building and deploying proprietary solutions that deliver value to millions of customers. You will partner with cross-functional teams to deliver AI-powered products, oversee the design, development, and deployment of AI software components, and make critical build-vs-buy decisions across a broad stack of AI technologies. This position offers the opportunity to invent and introduce state-of-the-art foundation model optimization techniques, contribute to the technical vision and roadmap of foundational AI systems, and attract and retain top talent in the AI industry.

$269k - $307k

Mercor is a leading AI data company that organizes human intelligence to power the AI economy. We build the layer between human expertise and frontier models, enabling millions of domain experts to train AI models. Our mission is to create a new category of work where expertise drives AI advancement, requiring an ambitious and fast-paced team working alongside researchers and AI companies at the forefront of technological change. Mercor is a profitable Series C company valued at $10 billion. As a Research Engineer Intern, you will work at the intersection of engineering and applied AI research. Your contributions will directly impact post-training, RLVR, synthetic data generation, and large-scale evaluation workflows for frontier language models. The work you do will be instrumental in training LLMs to master tool use, agentic behavior, and real-world reasoning in production environments. You will shape rewards, conduct post-training experiments, and develop scalable systems to enhance model performance, while also designing and evaluating datasets, creating data augmentation pipelines, and building rubrics and evaluators to push the boundaries of LLM learning.

Salary not listed

Together AI

We are seeking a reliability-minded software engineer to manage the production operations of our Kubernetes-native control plane, which provisions and runs our GPU inference fleet. This platform team develops the manifest-driven API and controllers that enable the inference team to define their needs, such as cluster configurations, model deployments, and capacity adjustments, and ensures these are reconciled. Your role will be to maintain the reliability of this system in production, addressing issues like InfiniBand problems during node communication, diagnosing faults in hero-model and LoRA serving that generic health checks miss, and enhancing automated self-healing to permanently resolve recurring failure modes. The primary goal is to ensure the decoupling remains effective even during failures and to maintain fleet efficiency without introducing new instabilities. You will be responsible for inference serving SLOs and error budgets, leading incident response, and implementing systemic fixes based on postmortem analyses. Additionally, you will contribute to the reliability aspects of capacity efficiency, recognizing that misconfigured defragmentation or bin-packing can pose production risks. We value candidates with hands-on production reliability experience in large-scale distributed or GPU-serving systems, proficiency in troubleshooting InfiniBand/RDMA and Kubernetes controller/CRD-based platforms, and a software engineering approach to GPU infrastructure and its supporting tools.

$160k - $280k

Figma is seeking a hands-on engineering leader to manage and grow its AI Infrastructure team. This team builds the foundational inference systems that power Figma's AI products, ensuring reliability, observability, and seamless integration of cutting-edge models. The role involves defining and executing technical strategy, collaborating across engineering, and directly contributing to the core AI platform. This is an opportunity to join a rapidly growing team at the ground floor and shape the future of AI at Figma.

$258k - $376k

NVIDIA is seeking a research engineer to join the Isaac team, focusing on physical AI for robotics. This role involves developing simulation tools for scientists and companies leveraging machine learning in robotics. You will bridge the gap between Isaac product engineering and NVIDIA Robotics research, transforming research concepts into product features and providing user feedback to the research loop. The team works at the intersection of simulation and machine learning, aiming to create the leading tool for evaluating robot foundation models in simulation, building on NVIDIA's expertise in photorealistic rendering and physics simulation.

$184k - $357k

Mercor is seeking an AI Infrastructure Engineer to build and manage the GPU and accelerator platform that powers our AI workloads. This role involves overseeing clusters, schedulers, training and inference serving, and the observability and cost data associated with fleet usage. You will write production code daily and own systems end-to-end, from design to on-call responsibilities. Mercor is a leading AI data company, building the layer between human expertise and frontier models, with a mission to organize human intelligence to power the AI economy. We are a profitable Series C company valued at $10 billion.

$200k - $400k

Figma is seeking a Support Engineer for its AI Infrastructure & Tooling team. This role will be the technical execution layer, bringing together support tools, customer context, internal systems, and AI workflows. You will design, build, and operationalize integrations across various systems to automate workflows, connect systems, and streamline support operations. The goal is to enhance the customer and Specialist experience by applying AI to improve support workflows, quality, and efficiency, ensuring the right context is available for issue resolution.

$169k - $245k

Together AI

The Developer Experience (DevX) team at Together AI is responsible for how developers learn and build on the platform, managing the documentation site, API reference, quickstarts, and cookbooks. They treat documentation as a product, with content living in Git and publishing through Mintlify, synchronized with the API via CI/CD automation. The team also focuses on making content accessible to AI coding agents, developing a house style guide for AI-assisted authoring and structured content. As a Technical Writer Intern on DevX, you will write, test, and ship documentation used by thousands of developers. You will be paired with a mentor and own significant projects from scoping to launch, with real users and metrics. This is a hands-on, technical role where you will run the code you document, call APIs, and contribute to the documentation repository from your first week. Potential projects include building migration kits, making documentation agent-ready, generating CLI reference from source, and writing evaluation design tutorials.

Salary not listed

Figma is seeking a versatile and experienced Machine Learning / AI Engineer to join our growing AI team, working at the intersection of applied machine learning, infrastructure, and product innovation. Whether you’re building intelligent search systems, crafting scalable data pipelines, or enhancing AI-powered creativity tools, your work will drive user productivity, shape new product experiences, and advance the state of AI at Figma. You’ll collaborate closely with engineers, researchers, designers, and product managers across multiple teams to deliver high-quality ML-driven features and infrastructure. This is a high-impact, cross-functional role where you’ll shape both foundational systems and user-facing capabilities. This is a full time role that can be held from one of our US hubs or remotely in the United States.

$153k - $376k

Figma is seeking infrastructure and platform-minded engineers to join our AI Platform team. This team builds the core frameworks, abstractions, and systems that power AI features across Figma, enabling new capabilities for AI product teams and improving performance, reliability, and technical quality. You will contribute to agent infrastructure, context retrieval & ranking platforms, and core AI services to accelerate critical company-wide AI initiatives. This role involves supporting end-to-end AI feature development by designing, building, and maintaining scalable, reliable, and performant systems, collaborating with cross-functional teams, and shaping platform strategy and best practices.

$153k - $376k

Figma is seeking a seasoned researcher to join its Agentic Experiences team. This role will focus on shaping and executing research to drive innovation in design and production tooling, particularly in the evolving landscape of AI-assisted product development. You will explore complex questions around trust, verification, and control in AI-driven experiences, influencing company strategy and product decisions. The ideal candidate is passionate about research, AI-led experiences, and translating user needs into impactful product strategies. This full-time position can be based in one of our US hubs or remotely within the United States.

$169k - $250k