Talent.com
Staff ML Infrastructure Engineer

Staff ML Infrastructure Engineer

Cubiq RecruitmentSonoma, CA, US
3 days ago
Job type
  • Full-time
Job description

Staff / Lead ML Infrastructure Engineer

San Francisco, CA — Onsite

Salary - Over market average + equity

We are building one of the world's leading generative video and multimodal AI platforms, and we're looking for a senior infrastructure engineer to drive the backbone that makes it possible. This role is ideal for an engineer from a top-tier tech company who has built cloud-scale systems, high-performance compute platforms, and battle-tested CI / CD pipelines that support complex ML workloads.

What You'll Own

  • Core ML Platform Architecture : Design and evolve the infrastructure that supports large-scale generative video and multimodal model training, evaluation, and deployment.
  • High-Throughput Compute Systems : Build and optimize GPU / TPU clusters, distributed training systems, and orchestration layers tailored for video-heavy pipelines.
  • Production Reliability for Generative Models : Create the tooling and services needed to safely push frequent model updates while handling massive compute loads and long-running jobs.
  • End-to-End CI / CD for ML : Lead the development of automated pipelines for model training, validation, artifact management, and production rollout.
  • Multimodal Data Infrastructure : Build systems to ingest, version, transform, and serve large-scale video, audio, and text datasets with high reliability.
  • Internal Developer Experience : Partner with research, product, and applied ML teams to build intuitive internal tooling for experiment tracking, model lineage, and resource scheduling.
  • Technical Leadership : Mentor engineers, set platform standards, and influence long-term architectural direction.

What You've Done

  • Experience architecting and operating large-scale infrastructure at a cloud provider, hyperscaler, or leading AI company.
  • Built or owned mission-critical CI / CD systems, high-capacity compute platforms, or data infrastructure supporting ML teams.
  • Deep experience with distributed compute across GPUs / accelerators, Kubernetes, and cloud infrastructure (AWS / GCP / Azure).
  • Strong engineering fundamentals in Python, Go, or equivalent languages.
  • Previous exposure to ML training pipelines—especially systems that handle heavy video, multimodal, or high-dimensional data.
  • Demonstrated ability to lead complex cross-org initiatives and drive technical strategy.
  • Nice to Have

  • Experience with video processing systems, large-scale media pipelines, or streaming architectures.
  • Familiarity with modern multimodal or video-generation frameworks (PyTorch, JAX, diffusers, custom accelerators).
  • Experience with Ray, Triton, CUDA optimization, or specialized scheduling for ML workloads.
  • Background working in high-growth AI startups or research-focused environments.
  • Security and compliance considerations for models that generate or process user content.
  • Why Join

  • Shape the underlying platform powering one of the most advanced generative video systems in the world.
  • Influence the future of multimodal AI by building infrastructure that directly accelerates research and product breakthroughs.
  • Work closely with experienced founding engineers, researchers, and platform builders from leading tech companies.
  • Highly competitive compensation, meaningful equity, and strong in-person engineering culture in San Francisco.
  • Create a job alert for this search

    Staff Engineer Infrastructure • Sonoma, CA, US

    Related jobs
    • Promoted
    Infrastructure Engineer

    Infrastructure Engineer

    FAR.AIBerkeley, California, United States
    Full-time
    AI is a non-profit AI research institute dedicated to ensuring advanced AI is safe and beneficial for everyone.Our mission is to facilitate breakthrough AI safety research, advance global understan...Show moreLast updated: 30+ days ago
    • Promoted
    Platform Infrastructure Engineer

    Platform Infrastructure Engineer

    Staffingine LLCConcord, California, USA
    Full-time
    Job Title : Platform Infrastructure Engineer.Design and implement cloud-native infrastructure on Google Cloud Platform (GCP) to support scalable secure workloads. Lead or contribute to cloud migrat...Show moreLast updated: 2 days ago
    • Promoted
    Staff Mechanical Engineer, Test InfrastructureReliability & Test • Berkeley, CA • Full time • On-site

    Staff Mechanical Engineer, Test InfrastructureReliability & Test • Berkeley, CA • Full time • On-site

    Form EnergyBerkeley, CA, United States
    Full-time
    Are you ready to build America's energy future? Form Energy is an American manufacturing and energy technology company.We're revolutionizing energy storage with cost-effective, multi-day technology...Show moreLast updated: 30+ days ago
    • Promoted
    Staff Application EngineerProject Engineering • Berkeley, CA; Somerville, MA • Full time • On-site

    Staff Application EngineerProject Engineering • Berkeley, CA; Somerville, MA • Full time • On-site

    Form EnergyBerkeley, CA, United States
    Full-time
    Are you ready to build America's energy future? Form Energy is an American manufacturing and energy technology company.We're revolutionizing energy storage with cost-effective, multi-day technology...Show moreLast updated: 30+ days ago
    • Promoted
    Staff Machine Learning Engineer, Optimization

    Staff Machine Learning Engineer, Optimization

    WaymoMartinez, California, United States
    Full-time
    Waymo is an autonomous driving technology company with the mission to be the most trusted driver.Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Wa...Show moreLast updated: 30+ days ago
    • Promoted
    • New!
    Manufacturing Engineer I (2nd Shift)

    Manufacturing Engineer I (2nd Shift)

    Collins AerospaceFairfield, CA, US
    Full-time
    HCA09 : Mission Sys (Propulsion) 3530 Branscombe Road , Fairfield, CA, 94533 USA.Person, or Immigration Status Requirements : . Ready to take your engineering career to new heights? Join our team at Co...Show moreLast updated: 4 hours ago
    • Promoted
    Construction Engineer

    Construction Engineer

    Halmar InternationalSonoma, CA, US
    Full-time
    Reporting to the Lead Construction Engineer, or delegate, the Construction Engineer is responsible for providing technical and engineering support during the planning of construction activities in ...Show moreLast updated: 2 days ago
    • Promoted
    Manufacturing Engineer II

    Manufacturing Engineer II

    Collins AerospaceFairfield, CA, US
    Full-time
    HCA09 : Mission Sys (Propulsion) 3530 Branscombe Road , Fairfield, CA, 94533 USA.Person, or Immigration Status Requirements : . Ready to take your engineering career to new heights? Join our team at Co...Show moreLast updated: 26 days ago
    • Promoted
    Senior Firmware EngineerSoftware Engineering • Berkeley, CA • Full time • On-site

    Senior Firmware EngineerSoftware Engineering • Berkeley, CA • Full time • On-site

    Form EnergyBerkeley, CA, United States
    Full-time
    Are you ready to build America's energy future? Form Energy is an American manufacturing and energy technology company.We're revolutionizing energy storage with cost-effective, multi-day technology...Show moreLast updated: 27 days ago
    • Promoted
    DevOps Engineer II

    DevOps Engineer II

    Kaseya CareersVacaville, California, United States
    Full-time
    IT infrastructure and security management solutions for Managed Service Providers (MSPs) and internal IT organizations worldwide powered by AI. Kaseya’s best-in-breed technologies allow organization...Show moreLast updated: 2 days ago
    • Promoted
    Casework and Millwork Designer

    Casework and Millwork Designer

    38° North Latitude Builders, Inc.San Rafael, CA, US
    Full-time +1
    Degrees North Latitude Builders, Inc.Degrees specializes in residential remodels and high end custom cabinetry.Extensive experience using Microvellum. Convert DWG files to DXF files.Ability to inter...Show moreLast updated: 30+ days ago
    • Promoted
    Senior Site Reliability Engineer

    Senior Site Reliability Engineer

    2kNovato, California, United States
    Full-time
    On-Call Requirement : Yes (Periodic Rotation).K is headquartered in Novato, California and is a wholly owned label of Take-Two Interactive Software, Inc. Founded in 2005, 2K Games is a global video g...Show moreLast updated: 30+ days ago
    • Promoted
    • New!
    Project Engineer

    Project Engineer

    Halmar InternationalSonoma, CA, US
    Full-time
    Reports to Project Manager or designated Deputy Project Manager.Major Duties & Responsibilities : .Manage all aspects of project documentation including submittals, RFIs and project meeting minut...Show moreLast updated: 12 hours ago
    • Promoted
    Staff Software Platform EngineerSoftware Engineering • Berkeley, CA; Somerville, MA; Weirton, WV • Full time • On-site

    Staff Software Platform EngineerSoftware Engineering • Berkeley, CA; Somerville, MA; Weirton, WV • Full time • On-site

    Form EnergyBerkeley, CA, United States
    Full-time
    Are you ready to build America's energy future? Form Energy is an American manufacturing and energy technology company.We're revolutionizing energy storage with cost-effective, multi-day technology...Show moreLast updated: 29 days ago
    • Promoted
    Structural Engineer

    Structural Engineer

    T&S StructuralSonoma, CA, US
    Full-time
    Structural Engineer (Levels 2–6).Employment Type : Full-time | On-site.T&S Structural provides comprehensive structural engineering services from conceptual planning through construction, ...Show moreLast updated: 30+ days ago
    • Promoted
    Traffic Control Technician II

    Traffic Control Technician II

    Roadsafe Traffic SystemsAmerican Canyon, California, US
    Full-time +1
    Title : Traffic Control Technician II • New employees start as Part-Time classification and promote to Full-Time classification as they work consistently Classification : Non-Exempt About the Organiz...Show moreLast updated: 23 days ago
    • Promoted
    Staff Commissioning EngineerProject Engineering • Berkeley, CA • Full time • On-site

    Staff Commissioning EngineerProject Engineering • Berkeley, CA • Full time • On-site

    Form EnergyBerkeley, CA, United States
    Full-time +1
    Are you ready to build America's energy future? Form Energy is an American manufacturing and energy technology company.We're revolutionizing energy storage with cost-effective, multi-day technology...Show moreLast updated: 30+ days ago
    • Promoted
    Foundation Inspector / Engineer

    Foundation Inspector / Engineer

    BEAR Engineering Inc.Sonoma, CA, US
    Full-time
    FEFF;Foundation & Drainage Inspector / Engineer (On-Site in the Bay Area, CA).BEAR Engineering provides independent, objective & data driven foundation, drainage, seismic & retaining wa...Show moreLast updated: 30+ days ago