Jobs › Companies › Waabi › Technical Lead Manager, ML Dataset Platform

Über diese Technical Lead Manager, ML Dataset Platform Stelle bei Waabi

Waabi · Hybrid · San Francisco, CA
Waabi, founded by AI visionary Raquel Urtasun, is the leader in Physical AI. With a world-class team, we're unlocking the next era of autonomous transportation with technology that's powering commercial autonomous trucks and robotaxis. Waabi is backed by and partners with world leaders in AI, automotive, logistics, and deep tech.
 
With offices in Toronto, San Francisco, Dallas, and Pittsburgh, Waabi is growing quickly and looking for diverse, innovative and collaborative candidates who want to impact the world in a positive way. To learn more visit: www.waabi.ai

As TLM of the ML Datasets Platform, you will play a critical role defining the APIs, tooling, and storage formats that make datasets a shared, first-class asset at Waabi: versioned, efficiently updatable, fully traceable, and fast to load across GPU workloads. The technical challenge is to find the right abstractions so that every ML team can integrate with the platform for its model development use case without being forced into a one-size-fits-all mold, and without building its own data infrastructure on the side. You will do this as a hands-on technical leader who also grows and manages the team, sets its roadmap, and partners closely with the ML Training Platform team and the ML development teams who are the platform's customers.
 

You will…

  • Lead and grow the ML Dataset Platform team as both its technical lead and people manager: set the technical direction, define the roadmap, hire, and mentor engineers.

  • Define the APIs, tools and storage formats for Waabi's ML datasets, balancing maximal reuse of data and tooling against the specific needs of each training workload.

  • Build the tooling to manage datasets at scale, including versioning and efficient incremental updates without full rebuilds.

  • Design the APIs that let teams express training-time transformations along the continuum between fully reproducible ahead-of-time ETL and dynamic transformation within data loaders.

  • Own the common dataloading tools used across workloads, including observability into bottlenecks and the throughput of training jobs.

  • Streamline data access for multi-cloud training so that datasets are available as needed with predictable cost and latency.

  • Make data lineage a first-class, fully automatic property of every dataset, from raw data through to the model trained on them.

  • Partner with the ML Training Platform team on the boundary between data and training infrastructure, and with the ML development teams on transitioning shared-dataset tooling into the new platform.

  • Drive adoption: work directly with model teams to migrate their training data onto the platform and feed what you learn back into your team’s roadmap.

 

Qualifications:

  • Deep ML dataset experience: You have designed and built large-scale data infrastructure for ML training (storage formats, dataset versioning, ETL, dataloading) and understand the performance characteristics of petabyte-scale sensor or multimodal data.

  • Strong software engineering and implementation: You have very strong Python skills and are comfortable in PyTorch training loops; you have strong software-engineering fundamentals and experience with distributed systems, cloud storage, and data processing frameworks (e.g., Spark, Ray, Beam).

  • Proven technical leadership: You have led a platform or infrastructure effort used by multiple teams, defined its APIs, and driven adoption across an organization.

  • People leadership: You have managed or tech-led a team of engineers, including hiring and mentoring, and can balance hands-on technical contribution with roadmap ownership and stakeholder management.

  • Team player: You communicate clearly with researchers and engineers, build consensus across teams with different needs, and deliver in a fast-paced environment.


Bonus:

  • Experience with ML data infrastructure in autonomous driving or robotics (multi-sensor logs, LiDAR, camera, HD maps).

  • Experience with columnar and tensor storage formats (e.g., Parquet, Lance, WebDataset, Zarr) and dataset versioning tools.

  • Experience with multi-cloud or hybrid training environments and the data-movement challenges that come with them.

  • Experience with data lineage, metadata catalogs, or feature-store systems.


The US yearly salary range for this role is: $200,000 - $280,000 USD in addition to competitive perks & benefits. Waabi US Inc.’s yearly salary ranges are determined based on several factors in accordance with the Company’s compensation practices. Note: The Company provides additional compensation for employees in this role, including equity incentive awards and an annual performance bonus.
 

Perks/Benefits:
  • Competitive compensation and equity awards.
  • Health and Wellness benefits encompassing Medical, Dental and Vision coverage (for full-time employees only).
  • Unlimited Vacation.
  • Flexible hours and Work from Home support.
  • Daily drinks, snacks and catered meals (when in office).
  • Regularly scheduled team building activities and social events both on-site, off-site & virtually.
  • As we grow, this list continues to evolve! 
 
Waabi is a technology start-up building technologies to transform the way the world moves. Join our talented team to be a part of the future and to make an impact!
 
Waabi is an equal opportunity employer. We celebrate diversity and are committed to creating a supportive, inclusive, and accessible workplace for all our employees. We seek applicants of all backgrounds and identities, across race, color, ethnicity, national origin or ancestry, age, citizenship, religion, sex, sexual orientation, gender identity or expression, military or veteran status, marital status, pregnancy or parental status, caregiver status, disability, or any other characteristic protected by law. We make workplace accommodations for qualified individuals with disabilities as required by applicable law. If reasonable accommodation is needed to participate in the job application or interview process please let our recruiting team know.
Bereit, sich bei Waabi zu bewerben?
Bei Waabi bewerben

Wie sich dieses Gehalt für Tech Lead vergleicht

Diese Stelle zahlt $240,000/yr — im Einklang mit der üblichen Spanne für Tech Lead Stellen.

$165,500 dem Median $215,500 $299,300

Übliche Spanne $176,899–$251,400/yr, aus 77 vergleichbaren Tech Lead Anzeigen auf JobsRadar (Vergütung auf USD hochgerechnet). Gehaltseinblicke für Tech Lead ansehen →

Ähnliche Jobs

Stitch Fix
Lead Engineer - Product Catalog Team
Stitch Fix
⚡ Früh bewerben Remote, USA · standortgebunden $125,000–$186,000
● Neu 👁 Gesehen ✓ Beworben vor 4 Std.
Fluidstack
PCS Lead Engineer, Power Generation
Fluidstack
⚡ Früh bewerben Austin, TX Vor Ort $263,000–$330,000
● Neu 👁 Gesehen ✓ Beworben vor 11 Std.
Fluidstack
BESS Lead Engineer, Power Generation
Fluidstack
⚡ Früh bewerben Austin, TX Hybrid $263,000–$330,000
● Neu 👁 Gesehen ✓ Beworben vor 11 Std.
Fluidstack
Talent Lead - Product and Engineering
Fluidstack
⚡ Früh bewerben Austin, TX Vor Ort $251,000–$311,000
● Neu 👁 Gesehen ✓ Beworben vor 11 Std.
Cartesia
Tech Lead, Product Engineering
Cartesia
⚡ Früh bewerben *HQ - San Francisco, CA Vor Ort $250,000–$375,000 🛂 Visa-Sponsoring
● Neu 👁 Gesehen ✓ Beworben vor 18 Std.
Kodiak
Tech Lead Manager, Motion Planning
Kodiak
⚡ Früh bewerben Mountain View, CA Vor Ort $245,000–$265,000 🛂 Visa-Sponsoring
● Neu 👁 Gesehen ✓ Beworben vor 21 Std.
Klaviyo
Lead Technical Project Manager, GTM Systems
Klaviyo
⚡ Früh bewerben San Francisco, CA Vor Ort $132,000–$198,000
● Neu 👁 Gesehen ✓ Beworben vor 22 Std.
Peregrine Technologies
Technical Product Marketing Lead
Peregrine Technologies
⚡ Früh bewerben San Francisco, CA Vor Ort $165,000–$225,000
● Neu 👁 Gesehen ✓ Beworben vor 1 Tg.
Peregrine Technologies
Applied AI Engineering Lead
Peregrine Technologies
⚡ Früh bewerben San Francisco, CA Vor Ort $225,000–$320,000
● Neu 👁 Gesehen ✓ Beworben vor 1 Tg.

Registrieren für Vorschläge, die auf die von Ihnen geöffneten Jobs und gespeicherten Suchen zugeschnitten sind.

Mehr Jobs bei Waabi

Alle Jobs bei Waabi ansehen →

Jetzt bewerben
🤖

Moment — langsam

JobsRadar wurde für echte Menschen gebaut, die eine schwere Zeit bei der Jobsuche haben — nicht für automatisierte Anfragen. Sie klicken viel zu schnell und sind jetzt vorübergehend blockiert.

Kommen Sie später wieder. Wenn Sie wirklich auf Jobsuche sind, stehen wir hinter Ihnen — verhalten Sie sich einfach wie ein Mensch.

Catch your next role the second it’s posted.

Create a free account and we’ll watch the boards for you — the instant a job matches your search, it lands in your inbox or Telegram. No digging, no refreshing.

Create free account

Free forever · takes 30 seconds · already have one?

Verschaffe dir einen Vorsprung bei der Jobsuche.

Tritt unserem Telegram-Kanal bei für das, was dir hilft, die Stelle zu bekommen — Gehaltsbenchmarks, den wöchentlichen Marktpuls und neue Feature-Drops. Kein Spam, nur Signal.

Dem Kanal beitreten — kostenlos