Traverse StaffingTraverse Staffing
← AI and data roles
AI and dataPermanentSunnyvale, US

Staff Software Engineer, AI Libraries

Staff engineer building Python libraries and platform tools for ML training and inference at a mid-sized software company in Sunnyvale.

About the role

A mid-sized software development company in Sunnyvale is hiring for this role. You will own the design and delivery of reusable Python libraries and platform tools that machine learning teams rely on across large GPU clusters. The role exists to improve reliability, performance and developer experience for those teams.

What you'll do

  • Design and deliver reusable Python libraries for multimodal data loading, distributed training and inference.
  • Build tooling for checkpointing and observability across large GPU clusters.
  • Lead projects end to end, from design through delivery.
  • Improve system reliability and performance.
  • Raise engineering standards and support machine learning teams.

What you bring

  • Strong Python skills, with experience building reusable libraries and platform tools.
  • Experience with distributed training and inference systems.
  • Familiarity with large GPU clusters, checkpointing and observability.
  • A record of leading projects end to end at a senior level.
  • A focus on developer experience and engineering standards.

Details

  • Location: Sunnyvale, California
  • Type: Full-time

Apply below and a Traverse recruiter will be in touch.

Apply

Two minutes. A real person reads it.

No resume needed at this stage: a LinkedIn profile and a few lines are enough. A recruiter calls before anything goes to a client.