Senior / Staff Software Engineer, ML Datasets & Data Pipelines
Posted Updated
Description
You will design and implement scalable ML data pipelines using driving and simulation data. You will optimize data formats, caching, and dataloading, improve sampling and data introspection, write tested code, and help plan, prioritize, and deliver projects.
Responsibilities
- Design and implement ML data pipelines for model training and evaluation
- Optimize data formats, caching, and dataloading
- Improve data sampling and composition for model-performance analysis
- Write high-quality, well-structured, rigorously tested code
- Support project roadmap planning, prioritization, and delivery
Requirements
- At least 4 years of industry experience
- Python proficiency
- Software engineering fundamentals
- Experience with PyTorch, TensorFlow, or JAX
- Experience building distributed ETL and data-processing pipelines
- Experience managing ML pipelines, dataset management, dataloading, and optimization
- Knowledge of cloud job orchestration, monitoring, and instrumentation
Benefits
- Equity awards
- Annual performance bonus
- Medical coverage
- Dental coverage
- Vision coverage
- Unlimited vacation
- Flexible hours
- Work-from-home support
- Daily drinks, snacks, and catered meals when in office
- Team-building activities and social events