AI Engineering Jobs in San Francisco

Figma is seeking PhD interns to join the AI Research team, focusing on the intersection of AI and Design. You will develop innovative AI solutions for design and creative products through fundamental and applied research, contributing to new AI-driven features. The team works on optimizing agentic systems, building robust evaluators, and post-training models, tackling challenging technical problems in AI and creativity. This internship offers the opportunity to publish research in leading AI/ML conferences.

$0k - $0k

Figma's AI Tools team is building AI-powered workflows, platforms, and tooling to enhance engineer productivity. This team owns systems that transform software development, including agentic CI auto-fixing, AI-assisted code review, and cloud agents that automate tasks like turning Slack messages into pull requests. The team operates across a three-layer platform stack: sandbox runtime, cloud agents, and workflow orchestration, while maintaining a growing portfolio of org-wide developer workflows.

$258k - $376k

Figma is seeking an Engineering Manager for Observability to lead a team responsible for building and evolving systems that monitor the behavior of Figma's platform and its AI products in production. This role involves owning the core observability stack (metrics, logs, distributed tracing) and the emerging AI observability pipeline, which processes traces, conversations, and model outputs to classify and evaluate AI features. The manager will set strategy, improve instrumentation and data quality, and explore AI-driven approaches for anomaly detection and operational automation. This is a unique opportunity to shape a critical, early-stage platform that underpins all engineering teams at Figma.

$258k - $376k

Figma is seeking founding Forward Deployed Engineers to join a new team focused on integrating Figma's products into large-scale enterprise environments. In this role, you will embed directly with strategic customers, working across their codebases, design systems, and workflows to ensure Figma's successful adoption within their unique technical constraints. You will tackle complex adoption challenges hands-on, from frontend stacks and legacy systems to security requirements, while building strong partnerships with customer engineering teams. This is a senior-level engineering position focused on customer success and product improvement, not sales or services. You will collaborate closely with Figma's Product, Engineering, Sales, and Customer Experience teams, influencing product development and the scaling of this new function. The role requires engineers who thrive in ambiguity, take end-to-end ownership, and are motivated by solving high-impact problems at the intersection of product, engineering, and customer needs. Periodic travel to customer sites is expected.

$153k - $376k

Figma is seeking a Data Platform Engineer to join their Data Engineering team. This role will focus on building the foundational systems that power Figma’s next generation of AI-driven products and data experiences. You will be an early engineer shaping the company's ML and data platform, enabling self-serve analytics and production-grade AI systems. The position involves working at the intersection of data, infrastructure, and machine learning to build scalable systems that empower Data Science, unlock AI capabilities, and bring data and models closer to the product experience.

$153k - $376k

Figma is seeking applied scientists with a Machine Learning and Artificial Intelligence background to build AI technologies and enhance Figma products. The role involves driving fundamental and applied research, combining industry best practices with a first-principles approach to design and build AI/ML models and systems. This is a full-time position that can be based in one of our US hubs or remotely within the United States.

$153k - $376k

Baseten

Baseten powers mission-critical inference for leading AI companies by uniting applied AI research, flexible infrastructure, and seamless developer tooling. We enable companies at the frontier of AI to bring cutting-edge models into production. The Agent Infra team builds a custom networking stack for our sandboxes, designed for the scale of numerous isolated machines that constantly suspend, resume, and exchange data. This role focuses on building a user-space networking stack using VPP, enabling multi-tenant virtual private networking for agents and services.

$160k - $330k

Sieve

Sieve is a multi-modal lab focused on curating high-quality training datasets across video, audio, images, text, and 3D, aiming to solve the bottleneck in the growth of AI applications. As a perception-focused Machine Learning Engineer, you will build systems that transform raw sensor footage into training data crucial for robotics and world-model teams. This role requires a strong background in computer vision, particularly in shipping perception systems for robots, devices, or autonomous vehicles, and applying these skills to massive datasets.

Salary not listed

The Machine Learning Center of Excellence (MLCOE) at JPMorgan Chase partners across the firm to shape, create, and deploy Machine Learning Solutions for challenging business problems. As a Quant AI Summer Associate within the MLCOE, you will join a team of technologists, ML researchers, and engineers exploring cutting-edge research and applying the latest Machine Learning techniques to J.P. Morgan’s unique data assets. You will collaborate directly with traders, portfolio managers, quants, data scientists, and salespeople to drive the data-led transformation of our businesses. Our Summer Associate Internship Program offers professional growth and development through project work, mentorship, and engaging speaker series, with potential for full-time employment offers upon successful completion.

Salary not listed

Databricks

The Web Platform team at Databricks builds and owns the public-facing web experiences that represent Databricks to the world. We are rebuilding the platform from the ground up, AI-native from the start, and pioneering an agentic SDLC as our operating model. As Staff Software Engineer on the Web Platform team, you will own the platform layer that the rest of the team builds on, including the web framework, design system, CMS, infrastructure, CI/CD pipelines, AI-enabled translation, and SDLC agents. This is a rare opportunity to design a web platform for an AI-native world from the ground up, with the north star architecture being defined now, and you will be a primary author of it. You will partner with technical leaders across web and marketing engineering, and set the standard for what AI-native platform engineering looks like in practice.

$198k - $273k

This customer-facing role involves owning the end-to-end technical relationship with enterprise clients, translating complex business challenges into high-impact AI solutions built on the Moveworks Platform. You will serve as the primary technical point of contact throughout the full implementation lifecycle, from solution design through launch, and will be comfortable collaborating with both customer IT teams and internal engineering and product teams. This is not a back-office role; you will be in front of customers regularly and are expected to travel up to 25% of the time.

$143k - $243k

Nebius is building a full-stack AI cloud platform to support developers and enterprises from data and model training through to production deployment. As a Senior Machine Learning Engineer on the Applied AI team, you will own model and endpoint optimization from model artifacts through production deployment. Your work will span model internals, quantization, model compression, speculative decoding, KV-cache optimization, inference engines, and serving architecture to improve latency, throughput, memory efficiency, GPU utilization, and cost per token while preserving model quality and reliability. You will collaborate with kernel and platform engineers to investigate serving problems, compare configurations, and resolve performance and quality regressions, validating improvements through reproducible benchmarks and safe production rollouts.

$195k - $262k

Scale AI

As a Forward Deployed AI Engineering Manager on the Enterprise team, you will serve as the technical bridge between Scale AI's advanced AI capabilities and strategic customers. This management role requires a blend of deep engineering and AI expertise, team leadership, and customer-facing problem-solving. You will collaborate with enterprise clients to understand their unique challenges, lead a team in architecting specific AI solutions, and ensure the successful deployment and adoption of AI systems within their production environments. You will work directly with customer engineering teams to integrate AI into their critical workflows.

$227k - $325k

OpenAI

OpenAI's Forward Deployed Engineering team partners with customers to turn research breakthroughs into production systems, operating at the intersection of customer delivery and core platform development. As a Forward Deployed Engineer for AI Infra, you will lead complex deployments of frontier models in production, embedding with customers where model performance and urgent delivery are critical. You will work directly with customer engineers, researchers, and product engineers on optimization problems across semiconductors and AI infrastructure, writing and debugging software, running experiments, and improving tools. Your role involves defining problems, structuring delivery, and shipping fast, with a focus on inference performance and chip-design optimization, ultimately contributing improvements to OpenAI's platform.

$185k - $300k

Brex

Brex is seeking a Staff Software Engineer to join their Product Data Platform (PDP) team. This role will focus on building and operating high-performance data, query, and search infrastructure that supports critical Brex product experiences, including AI-powered features. The PDP team's mission is to enhance Brex's customizability, scalability, and reliability for finance teams through deeply optimized, production-grade distributed systems. As a Staff Software Engineer, you will be instrumental in shaping the technical direction of the platform across multiple teams, identifying and resolving high-impact architectural challenges, and driving them to completion. You will be responsible for the group's technical quality, bridging long-term strategy with hands-on execution, and leading through influence.

$240k - $300k

SambaNova

The SambaStack team is seeking an Engineering Director to lead a software engineering team while remaining a hands-on technical contributor. You will own the growth and effectiveness of your team, the technical quality of their output, and the roadmap for the inference serving platform. You will continue to design systems and write code on the platform's hardest problems, contributing directly as a software engineer on critical platform initiatives.

$245k - $325k

RoboForce is an AI robotics company developing Physical AI-powered Robo-Labor for dull, dirty, and dangerous work. The company's robots are engineered for demanding industrial environments, with a focus on real-world deployment and scalability. We are looking for a Senior / Staff AI Research Engineer, Data Infrastructure to build the data and learning engine behind RoboForce's Physical AI stack. In this role, you will own the full pipeline — from raw teleoperation and UMI device data collection through curation, annotation, and storage, to post-training infrastructure that scores demonstrations, identifies failure patterns, and closes the loop back into model retraining.

$185k - $325k

Lila Sciences is building the infrastructure for autonomous science, creating AI systems that design and execute experiments across life sciences, chemistry, and materials science. The platform is designed for reliability and scale, serving external customers who require security, auditability, and operational rigor. This role focuses on enabling scientists and AI agents to query scientific knowledge, including literature, patents, customer documents, and experimental records, through advanced retrieval systems.

$224k - $336k

We are building AI that changes how people capture, understand, and act on conversations and meetings. As Director of AI Research, you will lead research and development across automatic speech recognition (ASR), speaker identification (SID), large language models (LLMs), and audio language models. Your work will shape our AI capabilities, product direction, and long-term research strategy. This is a hands-on leadership role for someone who combines research depth with experience building and deploying AI systems at scale. You will set technical priorities, guide model and infrastructure decisions, and work directly with the team on challenging research and engineering problems. Partnering closely with Product, Engineering, and Data teams, you will turn research advances into reliable product capabilities. You will also build the team, experimentation practices, and technical foundation needed to move quickly while maintaining high standards for quality and performance.

$244k - $313k

We are seeking a highly motivated and experienced Enterprise Data Management – Data Application Engineer specializing in Supply Chain to join our Business Applications group. This role is crucial in transforming computing by leading the creation, oversight, and advancement of enterprise data platforms and global business workflows. You will partner with business and IT teams to deliver scalable, resilient, and future-ready data solutions, leveraging your expertise in enterprise data management, supply chain operations, data architecture/platforms, and Agentic AI. Your contributions will be vital in resolving complex issues, refining data governance and observability, and ensuring trusted, high-quality information across the supply network.

Salary not listed

The AI Kernels & Compilers team at GM is at the forefront of developing advanced automated driving technologies, aiming for a future with Zero Crashes, Zero Emissions, and Zero Congestion. This role focuses on transforming cutting-edge AI research into production-grade software that runs efficiently and reliably on real vehicles. The team builds high-performance GPU kernels and custom libraries essential for on-vehicle ML inference in ADAS and autonomous driving systems, ensuring core AI workloads are faster, more reliable, and easier to deploy under real-world constraints.

$170k - $258k

Baseten

Baseten powers mission-critical inference for leading AI companies, enabling them to bring cutting-edge models into production. We are building the platform engineers use to ship AI products, recently securing $1.5B in Series F funding. The Agent Infra team is responsible for the core virtualization layer that powers our agent sandboxes. This layer provides each agent with its own isolated microVM, capable of running code, calling tools, and maintaining state across long tasks. We are building a cloud infrastructure from the ground up, focusing on multi-tenancy and deep hardware integration.

$165k - $330k

Nebius is building a full-stack AI cloud platform to support developers and enterprises in the global AI economy, from data and model training through to production deployment. The Nebius Token Factory is seeking scientists to tackle frontier LLM and VLM inference bottlenecks by turning them into research questions, conducting rigorous experiments, and translating findings into production capabilities. You will own focused research and optimization projects, write strong code, and produce prototypes, publications, and technical artifacts for engineers to build upon. Your research will cover areas like quantization, distillation, speculative decoding, KV-cache optimization, and model/runtime co-optimization, alongside inference engines and distributed inference architectures. You will evaluate trade-offs across model quality, latency, throughput, memory footprint, and cost per token, and collaborate with engineering teams to bring promising results into production.

$195k - $262k

SambaNova

SambaNova is seeking a Principal Engineer to own kernel-level drivers and low-level software for high-speed IO, memory, and networking subsystems within the DataScale® platform and RDU architecture. This role is central to delivering reliable, high-performance AI systems at scale. This is a hands-on Linux kernel and device driver role; it is not an application or platform software role. You will lead software bring-up of new silicon, deliver production drivers across the IO, memory, and network data paths, and debug complex issues spanning kernel, firmware, and hardware. This role suits an engineer with deep kernel and driver expertise and strong hardware fluency, who can operate in ambiguity and execute with high ownership.

$180k - $255k