Machine Learning Engineer Jobs in San Francisco

Applied Intuition

Applied Intuition is seeking a Senior Machine Learning Engineer to advance the core reconstruction technology behind their Neural Simulation product. This role involves transforming real-world driving data into high-fidelity simulation environments, pushing the boundaries of reconstruction quality, scaling data processing, and enhancing product utility for customers training autonomy systems. The engineer will directly influence the development of next-generation autonomous vehicles for major OEMs, working at the intersection of 3D computer vision, graphics, and machine learning to bring cutting-edge research into production and solve complex simulation challenges in Physical AI.

$190k - $260k

Applied Intuition

We are seeking senior machine learning engineers to advance the generative modeling technology behind our Neural Simulation product. This role involves pushing the boundaries of diffusion and video generation models to create realistic, controllable sensor data and augment real-world logs with new scenarios. Your work will directly influence the development of next-generation data-driven autonomous vehicles for leading OEMs. This position is ideal for engineers passionate about the intersection of generative modeling, computer vision, and machine learning, who are eager to bring cutting-edge research into production and address complex simulation challenges in Physical AI.

$185k - $260k

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 reinforcement learning methods for end-to-end driving models, identifying opportunities where learning from reward or feedback can improve beyond behavior cloning. You will take promising ideas from design through large-scale experiments, rigorous evaluation, and integration into their best driving models, contributing to the development of embodied intelligence that moves real vehicles safely.

Salary not listed

We are a dedicated research lab focused on building, understanding, using, and risk-managing foundation models. Our mission is to advance AI research, cultivate the next generation of AI builders, and make significant contributions to a knowledge-driven economy. As part of our team, you will work on the core of cutting-edge foundation model training alongside world-class researchers, data scientists, and engineers. You will tackle fundamental and impactful challenges in AI development, contributing to groundbreaking AI solutions that have the potential to reshape industries. Strategic and innovative problem-solving skills are essential for establishing our institution as a global hub for high-performance computing in deep learning.

$150k - $450k

We are seeking a Senior Software Engineer to join our team focused on enabling CUDA Math Libraries on new and specialized platforms. This role is crucial for ensuring that NVIDIA's math libraries deliver high functionality and performance across a wide range of targets, supporting the AI, data analytics, and scientific simulation industries. You will be responsible for the integration, functional qualification, compliance, and performance readiness of these libraries on emerging platforms, collaborating closely with core library and devops engineering teams. This is an exciting opportunity to contribute to the forefront of application performance in the AI era.

$152k - $288k

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 Core Model Safety team, which builds foundational capabilities for assisted and automated driving, focusing on collision avoidance, model understanding, and robustness under failure. You will work in a focused, high-impact senior team with strong ownership, access to large-scale training and fleet data, and close partners in research, simulation, evaluation, and applied engineering. As a Senior/Staff Machine Learning Engineer, you will lead the technical direction and delivery of learned emergency manoeuvre prediction and collision detection models. These are rare, high-consequence events that place unusual demands on data, modelling, and validation. You will take the program from problem definition through modelling, evaluation, integration, and evidence for deployment, driving core model safety roadmap themes and owning the full lifecycle from research to offline/online experiments to technology transfer.

$312k - $389k

We are a dedicated research lab focused on building, understanding, using, and risk-managing foundation models. Our mission is to advance research, nurture the next generation of AI builders, and drive transformative contributions to a knowledge-driven economy. As part of our team, you will have the opportunity to work on the core of cutting-edge foundation model training alongside world-class researchers, data scientists, and engineers. You will tackle fundamental and impactful challenges in AI development, participating in the creation of groundbreaking AI solutions that have the potential to reshape entire industries. Strategic and innovative problem-solving skills are essential for establishing our institution as a global hub for high-performance computing in deep learning.

$150k - $450k

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 on the Model Foundations team, which builds the geometric vision and 3D foundation models that underpin Wayve's autonomous driving systems. You will work at the intersection of large-scale deep learning, geometric computer vision, and real-world robotics to develop models that learn 3D structure and dynamics from fleet-scale sensor data. As a Principal Engineer, you will be a hands-on technical leader, setting direction for geometric vision, prototyping and training new model architectures, building the data and supervision needed to scale them, and taking successful ideas through to deployment on real vehicles.

$370k - $420k

We are a dedicated research lab focused on building, understanding, using, and risk-managing foundation models. Our mission is to advance research, nurture the next generation of AI builders, and drive transformative contributions to a knowledge-driven economy. As part of our team, you will work on the core of cutting-edge foundation model training alongside world-class researchers, data scientists, and engineers. You will tackle fundamental and impactful challenges in AI development, contributing to groundbreaking AI solutions that have the potential to reshape entire industries. Strategic and innovative problem-solving skills are essential for establishing our institution as a global hub for high-performance computing in deep learning.

$150k - $450k

Wayve is seeking a Robotics Software Engineer to join our Online Calibration team and support scaling across our autonomous vehicle fleet. You will turn calibration and estimation methods into reliable, observable, and resource-efficient components of the on-vehicle runtime. Working across calibration, state estimation, robot software, hardware acceleration, and vehicle-platform teams, you will help shape the architecture and roadmap for a foundational capability that directly affects vehicle performance, resilience, and safety. Your day-to-day will involve developing robust production software for online calibration, operating continuously on constrained, hardware-accelerated platforms, and working with asynchronous sensor data, accelerator-produced features, and noisy, time-varying calibration parameters, balancing correctness, debuggability, and runtime efficiency.

$276k - $294k

Wayve is building the AI platform for autonomous driving, enabling vehicles to learn directly from real-world experience. This role is on the Enrichment and Curation team, which transforms fleet-scale driving data into high-quality training datasets, enrichments, and evaluation benchmarks for the company's end-to-end AI Driver and foundation models. You will be instrumental in developing the embodied intelligence that moves real vehicles safely and the ecosystem that future machines will run on.

$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 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 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

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

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

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 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

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

SambaNova

As a Senior Principal Machine Learning Engineer, you will be responsible for designing, developing, and optimizing machine learning models, with a focus on cutting-edge Large Language Models (LLMs), to run efficiently on SambaNova's specialized hardware architecture, including the RDU. This critical role bridges advanced LLM research and practical deployment, involving the development of model architectures, improving training and inference efficiency, and collaborating on hardware-software co-design with compiler, systems, and hardware teams. The engineer will also act as the ML expert, guiding the integration of LLM solutions into production systems and customer-facing products like SambaStack and SambaCloud, with work spanning the full lifecycle from training and inference to evaluation and data curation.

$220k - $300k

SambaNova

SambaNova is seeking a Compiler Architect to join their team and drive innovation in compiler infrastructure and optimization algorithms. This critical role involves working with various layers of the compiler stack and coordinating with development teams to ensure state-of-the-art ML model performance on the SambaNova platform. Responsibilities include analyzing PyTorch and machine learning models to map operations onto the company's unique dataflow hardware.

$210k - $280k

OpenAI

We are seeking a software engineer to build and evolve our classification and evaluation platform. You will develop services, online and offline pipelines, and tools that help researchers and data scientists turn an idea into a working measurement and iterate on it quickly. This role combines software and data engineering with hands-on production ownership. You will shape the architecture, deliver new capabilities, improve inference efficiency, and keep the system dependable as workloads and requirements evolve. Strong candidates will bring deep software engineering expertise and an interest in building a technical product in close partnership with its users.

$347k - $445k

Anthropic

The Account Abuse team is responsible for ensuring Anthropic's computing capacity is allocated fairly by minimizing resources for bad actors and preventing their return. As a software engineer on this team, you will build the machine learning systems that detect and stop abuse at scale. The ideal candidate can anticipate opponent strategies and responses to countermeasures. This role involves full-stack machine learning engineering across model training, productionization, and evaluation, with opportunities to leverage Claude to accelerate model development and maintenance. The focus is on classical ML applied to structured and behavioral data, emphasizing robust production systems, precise measurement, and safe rollout strategies, as false positives can lead to legitimate customer lockout.

$320k - $485k

Latent is seeking a Machine Learning Engineer, Research, to lead the design and development of innovative modeling approaches that push the boundaries of clinical intelligence. This role involves driving research from initial problem definition to validated outcomes, influencing how models learn from longitudinal patient data. We are looking for senior and staff-level engineers adept at managing critical research challenges from inception to completion, with a focus on problems that directly impact patient well-being.

$225k - $300k