Machine Learning Engineer Jobs in San Francisco

Glean is seeking a Machine Learning Engineer to enhance the quality of its AI Assistant and autonomous agents. This role bridges production machine learning, LLM-powered systems, and product engineering, focusing on developing, evaluating, and refining assistant experiences for reliability and usefulness within enterprise workflows. You will tackle applied challenges in agent quality, evaluation, personalization, retrieval, and orchestration, with an emphasis on shipping production systems and improving the assistant's capabilities over time through enhanced signals, feedback loops, and end-to-end execution.

$180k - $205k

SpaceXSI is building Super Intelligence to help humanity understand the universe, creating a powerful flywheel where more compute trains better models, better models power more useful products, and what we learn from people using those products shapes what we build next. You will own and elevate our mission-critical research platform that directly accelerates SpaceXSI's frontier work. The evals team creates and executes instruments for measuring model capabilities and behaviors, used to clean datasets, provide feedback for experiments, and track growth throughout training. We also diagnose why agents fail and how to make improvements, managing execution infrastructure, designing data sets and grading schemes. You will design, build, and ship delightful and bulletproof systems that engineers love to use every day, valuing strong attention to detail, first principles reasoning, and great taste.

$175k - $275k

Vercel

Vercel is seeking a Machine Learning Engineer for its Trust & Safety Engineering team. This role focuses on building and operating production systems to detect and prevent platform abuse at scale. It's an engineering-intensive position, with approximately 80% of the time dedicated to writing and deploying production code, including services, pipelines, and infrastructure, and 20% focused on modeling. The goal is to take detection and classification approaches from initial prototypes to robust, production-ready systems.

$208k - $312k

Stripe

Stripe is seeking PhD machine learning engineering interns for the summer of 2027 to contribute to critical projects that enhance Stripe's products. Interns will focus on areas such as foundation models for fraud detection, enhanced support, and user behavior prediction. This role offers an exceptional opportunity to work on challenging problems at the intersection of finance, technology, and data, including creative projects like the Stripe Assistant and the Stripe Foundation Model, leveraging machine learning to revolutionize business interactions with financial services.

Salary not listed

CoreWeave

CoreWeave is seeking a Senior Engineer for its Benchmarking & Performance team. This role will be instrumental in managing a planet-scale performance data warehouse, handling the ingestion, storage, transformation, and analysis of performance events across global data centers. The engineer will also contribute to industry-leading performance benchmarking publications like MLPerf. The position involves owning designs, elevating engineering standards, and driving measurable improvements in latency, throughput, and reliability across multiple services. Collaboration with product, orchestration, and hardware teams will be key to evolving the Kubernetes-native platform and meeting stringent P99 SLAs at scale.

$182k - $242k

Scale AI

Scale is at the forefront of the AI revolution, developing data engines and technologies that power the world's leading LLMs and generative models. This role is on the Platform Engineering team, responsible for the foundational data infrastructure that supports these cutting-edge AI products. You will lead the design and development of core data storage, streaming, caching, and indexing platforms, gaining exposure to the rapidly evolving AI landscape across various industries. The work involves driving architecture, implementation, and reliability of these critical systems, collaborating with stakeholders, and mentoring junior engineers.

$252k - $315k

NVIDIA is seeking a Senior MLOps Engineer to join our DSX Enablement team, collaborating closely with strategic customers to implement and enhance groundbreaking AI workloads. We partner with the world's most innovative AI companies and open-source communities to address their most challenging technical problems. In this role, you will develop innovative solutions that advance AI infrastructure capabilities, advise infrastructure experts on the demands of ML workloads, and help practitioners diagnose and solve full-stack AI and ML system problems. You will work on a team with direct responsibility for the success of internal and external customers’ AI and ML initiatives, including LLM performance evaluation and supporting new hardware in open-source frameworks.

$184k - $357k

As an ML Infrastructure Engineer, you will play a pivotal role in building and optimizing the reliable, high-performance ML platform that powers recommendations on X. We're looking for exceptional engineers who are passionate about our mission and have a strong desire to make a meaningful impact.

$180k - $440k

Stripe

Stripe is seeking Master's machine learning engineering interns for the summer of 2027. This internship offers a unique opportunity to contribute to critical projects that enhance Stripe's products, with a focus on foundation models for tasks like fraud detection, customer support, and user behavior prediction. Interns will tackle complex challenges at the intersection of finance, technology, and data, working on innovative projects such as the Stripe Assistant and the Stripe Foundation Model to transform how businesses engage with financial services and data.

Salary not listed

CoreWeave

CoreWeave is seeking a Senior Engineer for its Benchmarking & Performance team, focusing on GPU kernel authoring and optimization. In this role, you will be responsible for writing, profiling, and tuning GPU kernels that are critical for large-scale model serving, aiming to maximize throughput and minimize latency. You will also contribute to industry-leading performance benchmarking publications. This position requires an individual who can lead designs, improve engineering standards, and deliver significant enhancements to latency, throughput, and reliability across the inference stack. You will collaborate with product, orchestration, and hardware teams to translate kernel-level optimizations into system-wide improvements and meet stringent Service Level Agreements (SLAs) at scale.

$182k - $242k

NVIDIA is building the next generation of AI-native autonomous driving architecture, integrating classical safety stacks, foundation models, and scalable AI systems into a unified production platform. This role is for individuals passionate about shaping the future of autonomous systems over the next decade. The work involves the deep integration of classical safety architectures and large-scale AI-driven systems, moving beyond purely classical robotics or end-to-end neural networks.

$224k - $431k

CoreWeave

CoreWeave is seeking a Principal Software Engineer to own the frontend architecture of the W&B app, which is used daily by ML researchers and engineers. The role focuses on ensuring app performance, stability, and reliability by addressing challenges in data composition and load times. This is a frontend-led position requiring deep architectural understanding and cross-organizational collaboration. You will be responsible for defining architectural boundaries, shaping designs across teams, and executing plans to improve the user experience and product features.

$227k - $303k

Join NVIDIA's global Developer Technology (DevTech) team as an AI Developer Technology Engineering Manager. In this role, you will lead a team of highly skilled engineers focused on accelerating the end-to-end performance of real-world Deep Learning and Machine Learning applications. You will also develop novel algorithms to optimally leverage NVIDIA hardware, working on applications such as Large Language Models (LLMs), Computer Vision, Speech, Recommender Systems, and Multimodal architectures. As a leader, you will define and drive strategic technical initiatives, setting goals and collaborating closely with customers, NVIDIA Research, and hardware/software teams to advance the state-of-the-art in accelerated AI.

$224k - $357k

CoreWeave

CoreWeave is seeking a Principal Engineer to lead its Benchmarking & Performance team. This role will be responsible for a global performance data warehouse, ingesting, storing, transforming, and analyzing performance events across all data centers. You will also play a key part in industry-leading performance benchmarking publications, working closely with NVIDIA and the open-source community to demonstrate CoreWeave's performance and reliability leadership in AI.

$206k - $333k

NVIDIA is seeking a Deep Learning Communication Architect to join its software architecture group. This role involves scaling Deep Neural Network (DNN) models and training/inference frameworks to systems with hundreds of thousands of nodes. You will be instrumental in optimizing communication performance by identifying and eliminating bottlenecks in data transfer and synchronization. The position requires designing efficient communication protocols tailored for deep learning workloads and collaborating with hardware and software teams to craft systems that leverage high-speed interconnects and communication libraries. You will also research and evaluate new communication technologies to enhance performance and scalability, and develop proofs-of-concept to validate new strategies.

$184k - $357k

We are seeking software engineers to design and develop libraries and tools that simplify and accelerate computing for unstructured sparsity in Deep Learning (DL) and High-Performance Computing (HPC). Our team develops GPU-accelerated libraries and SDKs that are crucial for AI, data analytics, and scientific simulations. In this role, you will contribute to solutions involving sparse tensor computations, domain-specific language (DSL) specifications for sparse storage formats, and on-demand code generation. The ideal candidate will have experience in accelerated computing software development and a passion for advancing the state-of-the-art in various accelerated computing domains and DL frameworks like PyTorch.

$184k - $357k

Nvidia is seeking a Distinguished Engineer for Wireless Infrastructure to lead technology strategy for next-generation wireless infrastructure, focusing on AI-RAN and Agentic Core. The role involves applying AI/ML to 6G radio access networks (RAN), transforming the wireless core into an agentic AI-based packet core, and driving rapid prototyping of GPU-accelerated algorithms. You will collaborate with industry partners and customers to bring innovations to market and work with standards bodies like 3GPP and O-RAN to integrate AI-native approaches into future technologies. The position also includes driving development for new applications in Integrated Sensing and Communications (ISAC), Semantic Communications, and Physical AI at the Edge.

$320k - $489k

NVIDIA is seeking architects to design hardware accelerator and processor architectures that enable state-of-the-art machine learning and data analytics algorithms and applications on next-generation mobile, embedded, and datacenter platforms. This role offers the opportunity to make a real impact in a dynamic, technology-focused company. As a member of the deep learning architecture team, you will contribute to features that advance the state of AI on next-generation GPUs and systems. This position requires staying current with the latest DL research and collaborating with diverse teams, including DL researchers, hardware architects, and software engineers.

$224k - $357k

NVIDIA Research is seeking exceptional systems researchers to contribute to the development of hardware, software, and infrastructure technology for ML systems of all scales. Advances in AI/ML heavily rely on the development of efficient, scalable, resilient, and trustworthy systems for training, fine-tuning, and serving ML models. AI/ML applications are also pushing the limits of both personal devices and warehouse-scale data centers. All layers of AI systems need to be co-designed and co-optimized to maximize performance and energy efficiency, improve scalability, and support emerging algorithms in this space. We are seeking candidates that have a proven track record of research excellence, a broad perspective on systems design across the full hardware/software/infrastructure stack, and depth in one of the layers such as AI/ML systems, operating systems, distributed systems, data management, cloud computing, and computer architecture.

$168k - $265k

NVIDIA is seeking engineers to develop algorithms and optimizations for our LPX inference and compiler stack. You will work at the intersection of large-scale systems, compilers, and deep learning, crafting how neural network workloads map onto future NVIDIA platforms. This is your chance to be part of something outstandingly innovative!

$152k - $288k

NVIDIA's DGX Cloud team is seeking a Senior Full-Stack Software Engineer to join the AI Hub team. This role will accelerate AI research by ensuring NVIDIA's AI infrastructure is used efficiently, transparently, and at scale. The primary goal is to build a unified, self-service portal for AI researchers to manage, monitor, and optimize their use of Managed AI research Superclusters. You will lead the architecture and delivery of high-scale web products, own multi-team initiatives, and drive reliability and performance improvements.

$224k - $357k

NVIDIA is seeking a Senior Robotics Research Engineer to join its Seattle Robotics Lab, focusing on AI and robotics for drug discovery. This role involves developing fundamental robotics technology and building real-world systems to automate and bring autonomy to molecular discovery, manufacturing, testing, and validation. You will work on creating digital twins of robots and lab environments, simulating robot physics, developing perception pipelines, and translating experimental protocols into executable physical procedures. The position requires training robots for complex manipulation tasks and integrating these systems into biological experiment workflows, collaborating with research scientists and engineers across NVIDIA and Eli Lilly.

$184k - $288k

Join the CPU performance architecture team to drive the development of CPU technology for diverse applications including AI/deep learning, high-performance computing, cloud services, gaming, virtual reality, and autonomous vehicles. This role involves pushing performance boundaries for all CPU products by analyzing and optimizing workloads across various markets and identifying critical application behaviors. You will work with architects to improve future CPU and system designs and benchmark NVIDIA's CPU offerings against competitors.

$224k - $431k

We are seeking a Senior Performance Modeling Architect to define and enhance our next-generation CPU Cache Hierarchies and interconnects. This role offers a unique opportunity to develop scalable solutions that bridge the high-reliability, low-latency requirements of Automotive systems with the efficiency and density demands of Data Center environments. You will be instrumental in creating the core models that manage data flow across our silicon, ensuring our L3/System Caches and coherent fabrics meet ambitious performance targets.

$152k - $288k