Machine Learning Engineer Jobs in San Francisco

Lila Sciences is building a platform where AI and automation co-evolve to solve complex problems in medicine. The Foundation Models team within Life Science AI (LSAI) focuses on developing foundation models that integrate biological sequence, molecular structure, and experimental data to drive automated scientific discovery. This role involves contributing to problem formulation, model design, training, evaluation, and integration into Lila's discovery engine, with a specific emphasis on structure prediction and co-folding for antibody and biologics design. The position is for an individual contributor focused on building deep expertise in structure-aware generative AI for biology, owning research sub-problems, and collaborating with experimental scientists to close the computational-experimental loop.

$176k - $304k

NVIDIA is seeking a graduate research intern to advance integrated sensing and communications (ISAC) on GPU-accelerated platforms. The intern will join a team building AI-native wireless systems and develop, train, and test ISAC algorithms end-to-end on the NVIDIA Aerial Testbed. This role is ideal for a student passionate about deep algorithm work and translating ideas into functional systems.

$0k - $0k

Athelas

Commure is building the AI Operating System for healthcare, transforming the industry's administrative burden with an AI-native platform. Our products span ambient AI for documentation, intelligent agents for workflows, and autonomous revenue cycle management. We are seeking a Senior Engineering Manager to lead the team responsible for the financial core of our revenue cycle product, including the ledger, reporting, and analytics. This role involves managing a team of 6-8 engineers, owning the product roadmap with a product partner, and ensuring financial correctness across the platform. The team is currently undertaking a significant rebuild of its reporting logic and ledger system while maintaining service for existing customers.

$240k - $275k

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. The Driving Product team specifically focuses on developing the AI driver's core behaviors, taking models from data and training through evaluation and on-road validation, and shipping them into cars on aggressive timelines. This role involves owning key parts of the training model lifecycle, from data curation to iteration planning, and developing AI driver model architectures and training algorithms to enhance driving behaviors for various levels of autonomy.

$312k - $351k

NVIDIA is seeking a Staff Engineer to develop the company's Enterprise Data Governance platform. This platform is crucial for protecting sensitive information, managing data access across all collaboration and content systems, and building a knowledge infrastructure that ensures the trustworthiness and accuracy of enterprise AI. The role presents a unique challenge in balancing the need for broad data access for AI agents with strict security, legal, and export control requirements. This position is ideal for technically strong engineers at the senior to principal level who are passionate about solving complex infrastructure problems and contributing to NVIDIA's core operations at scale.

$168k - $322k

Salesforce is seeking a Senior Performance Engineer to join their Agentforce Operations team. This role is crucial for optimizing the scalability, resilience, and performance of their AI-powered supply chain platform. You will gain a deep understanding of the architecture and implement end-to-end optimizations to enhance scalability and reliability. The position involves tackling complex challenges such as optimizing data models, streamlining asynchronous operations, profiling infrastructure, and developing tools for experimentation and validation. As a subject matter expert, you will operate with autonomy, build alignment across engineering teams, and mentor other engineers, contributing to the company's ambitious scaling goals.

$149k - $224k

OpenAI

We are seeking a Machine Learning Engineer to develop and productionize new classification and evaluation capabilities. This role involves translating open-ended questions about model behavior into well-defined ML problems, building effective methods, and rigorously assessing their reliability. You will own classifiers throughout their lifecycle, from supervised fine-tuning and distillation to evaluation, deployment, and production inference. Collaborating closely with researchers to identify valuable problems and with software engineers to scale promising approaches will be key to success.

$381k - $510k

Wayve is building the leading AI platform for autonomous driving, developing an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is on the Driving team, which owns the core end-to-end driving model that directly shapes how Wayve's vehicles behave on the road. The team iterates quickly, identifying and resolving issues through data, model, and evaluation changes, and is central to Wayve's Robotaxi launches. This position offers a unique opportunity to lead a small team while remaining deeply technical, contributing to the development of embodied intelligence for real vehicles.

$370k - $395k

Dodo

Dodo is seeking a Senior ML Engineer to enhance the intelligence, speed, and reliability of healthcare's front desk agents. This role involves owning the models, evaluation, and training loops that process millions of real conversations daily, continuously improving the system based on real-world interactions. You will be instrumental in making Dodo's agents smarter and more effective in handling patient communications.

Salary not listed

Databricks

Databricks is seeking a Senior Engineering Manager for Search to lead the development and scaling of their next-generation Search product. This role is critical in building and evolving a highly performant, cost-efficient, and developer-friendly Search stack that enables customers to build AI applications with accuracy, speed, and scale. The manager will drive innovation across Vector Search, Keyword Search, and Search Quality, and will be responsible for shaping the long-term vision, mentoring engineers, and collaborating across teams to expand the impact of Search at Databricks.

$222k - $300k

NVIDIA is seeking strategic, ambitious, hard-working, and creative individuals for their 2027 Autonomous Vehicles and Robotics internships. This 12-week full-time program offers an excellent opportunity to expand your career and gain hands-on experience with industry-leading teams. Interns will work on projects with a measurable business impact, contributing to NVIDIA's pioneering work in AI and digital twins that is transforming major industries.

$0k - $0k

As an engineer on the Data Team, you will operate as a technical leader on the foundational layer of the platform. This team owns Qualified's data end-to-end, including leads and accounts data models, ETL pipelines, Data 360, Salesforce sync, analytics and insights, data integrations, and Data APIs. Every product area and AI agent depends on this layer being correct, fresh, and fast. This is a newer team with significant greenfield opportunities, where you will shape the architecture of core pipelines and data models, set patterns for other teams, and transform inherited systems into a coherent platform. You will also directly contribute to the intelligence of AI agents by ensuring the data they rely on is accurate and flows seamlessly between systems.

$173k - $260k

Databricks

Databricks' Model Serving product offers enterprises a unified, scalable, and governed platform for deploying and managing AI/ML models, including traditional ML and large language models. It provides real-time, low-latency inference, governance, monitoring, and lineage, making it a core pillar for customers operationalizing models at scale with strong SLAs and cost efficiency. As a Senior Engineer, you will be instrumental in shaping both the product experience and the foundational infrastructure of Model Serving. You will design and build high-throughput, low-latency inference systems for CPU and GPU workloads, influence architectural direction, and collaborate with cross-functional teams to deliver a world-class serving platform.

$166k - $225k

NVIDIA is seeking ambitious and creative individuals for its Spring 2027 Developer and Performance Technology internships. These 8-12 month full-time internships offer a unique opportunity to gain hands-on experience with industry-leading Deep Learning teams, working on projects that have a measurable impact on NVIDIA's business. Interns will tackle complex challenges in areas such as AI, digital twins, gaming, robotics, self-driving cars, and virtual worlds. The program is designed for students pursuing a B.S. or M.S. degree in a relevant field, providing a chance to expand their careers and contribute to transformative technologies.

$0k - $0k

NVIDIA is seeking strategic, ambitious, hard-working, collaborative, and creative Ph.D. Engineering interns to join their Computer Architecture and Systems teams for Summer 2027. This internship offers an excellent opportunity to expand your career and gain hands-on experience with industry-leading technologies. Interns will design and implement novel ideas in GPU and CPU architectures, systems architectures, operating systems, AI systems, and distributed systems, with the potential to transfer their research to product groups, resulting in prototypes, patents, products, or published research.

$0k - $0k

NVIDIA is seeking interns to research scalable differentiable algorithms, including adjoint and automatic differentiation methods. This role will involve connecting simulations with experimental data for calibration and data assimilation using AI/ML techniques. The work will be evaluated on simulation workloads and assessed for integration into NVIDIA software. This internship is ideal for individuals passionate about mathematics, code, hardware, and engineering applications.

$0k - $0k

We are seeking an exceptional hands-on engineer to help our partners and customers make demanding physics and robotics simulation workloads run exceptionally well on NVIDIA platforms. You will work at the intersection of algorithms, mathematics, GPU architecture, and production software to find the true limits of real applications and turn that understanding into durable code and actionable feedback for NVIDIA engineering teams. This role is for someone with a recognizable technical body of work: not someone who merely used advanced systems, but someone who took stewardship of one and made it substantially better.

$224k - $431k

Physical Intelligence

Physical Intelligence is building a platform to bring general-purpose AI into the physical world, powering robots with foundation models and learning algorithms. As an API Product Engineer, you will develop the core product surface that enables external companies to utilize Pi's models similarly to how they use LLM APIs today. This involves allowing them to bring their own data, fine-tune models, evaluate performance, and run low-latency inference within their own environments. You will own this entire experience, transforming close-knit collaborations into a scalable platform designed to support a vast number of robots.

Salary not listed

NVIDIA is seeking a Senior Synthetic Data Engineer to join the DRIVE team and contribute to automotive innovation. This role involves collaborating with technical leaders in autonomous driving and simulation to design and develop a simulation environment that advances autonomous vehicle technology. The position focuses on creating synthetic data crucial for training deep learning models used in autonomous driving systems, enhancing the reliability and speed of vehicle development through advanced simulation techniques.

$184k - $357k

Handshake is seeking a Machine Learning Engineer I for the Growth Relevance team to help shape the evolution of careers in the AI economy. This role will focus on developing, deploying, and enhancing machine learning systems for lifecycle optimization, personalized notifications, and monetization strategies. You will work with cutting-edge ML infrastructure, including embedding-based retrieval, Graph Neural Networks, and multi-stage rankers, contributing to critical marketplace metrics and responsible AI practices.

$150k - $189k

Luma is seeking a Research Scientist/Engineer to focus on optimizing the performance of their multimodal models. This role involves profiling and enhancing GPU, CPU, and accelerator code to ensure efficient training and scalable deployment without compromising quality. You will be responsible for writing kernels and operations that maximize hardware utilization and minimize latency, requiring deep expertise in GPU optimization, transformer internals, and low-level programming.

$195k - $395k

Metriport is revolutionizing US healthcare by solving the critical problem of medical data exchange. We transform unusable raw data from legacy systems into a clean, unified format that care teams and their agents can use to improve patient outcomes. Our technology enables rapid data retrieval, providing clinicians with complete patient histories instantly, which can lead to earlier diagnosis and treatment. We are a technology company building every layer of our platform, from data connection and transformation to generating insights, aiming to become the essential infrastructure for US healthcare. Having secured significant funding and achieved product-market fit with millions in ARR and over 100 customers, we are poised for substantial growth.

$200k - $260k

Knowtex is seeking an ML Engineer / Researcher focused on Large Language Models to contribute to the next generation of their AI stack. This role involves developing and optimizing models for clinical documentation and structured clinical reasoning, aiming to enhance quality, cost, latency, and control. The position is research-heavy with a direct path to production, requiring the design of experiments, building datasets and evaluation systems, and fine-tuning and post-training models. The successful candidate will collaborate closely with engineering and clinical teams to deploy solutions at scale and will play a key role in shaping Knowtex's long-term LLM strategy.

Salary not listed

Crusoe

Crusoe is seeking a Staff Software Engineer specializing in Linux Kernel Security to strengthen the security of their AI cloud infrastructure. This role involves implementing kernel security enhancements, triaging CVEs, securing the boot path, and establishing CI guardrails. You will research vulnerabilities, advocate for secure coding practices, and conduct security-focused code reviews. The position also includes utilizing and developing security auditing tools, supporting incident response, and balancing security with system performance. A key aspect is contributing to the security of Cloud Hypervisor and advancing confidential computing primitives. This role requires close collaboration with other engineering teams and providing technical leadership and mentorship.

$215k - $260k