Senior Software Engineer
Machine Learning Infrastructure
Posted on 2/7/2024
StackAV

51-200 employees

Company Stage

Series C

Total Funding

$1B

Founded

2023

Headquarters

Pittsburgh, Pennsylvania

Growth & Insights
Headcount

6 month growth

103%

1 year growth

103%

2 year growth

103%
Locations
Remote in USA
Experience Level
Entry
Junior
Mid
Senior
Expert
Desired Skills
Python
Airflow
Pytorch
AWS
NumPy
CategoriesNew
AI & Machine Learning
Software Engineering
Requirements
  • Experience in ML training (local and distributed), feature extraction, dataset creation
  • Strong programming skills in Python and C++
  • Familiarity with PyTorch, Numpy
  • Experience with large scale distributed systems and pipelines
  • Experience with cloud environment, storage and database management systems (AWS S3, Parquet)
  • Experience with workflow systems such as Apache Airflow
Responsibilities
  • Collaborate closely with ML modeling teams across Perception, Motion Planning, and other ML teams to define ML infrastructure requirements and build critical data services
  • Design and develop large scale data pipelines that transform sensor data from vehicles into consumable tool formats for ML modeling teams
  • Leverage MLOps best practices to design and develop scalable model training, evaluation, experimentation and deployment workflows
  • Use the latest cloud technologies to design, implement, and test scalable distributed data systems
  • Build high impact software and tools to move mission critical ML applications forward
Desired Qualifications
  • Understanding of ML modeling and concepts around train/test/validation and ML modeling lifecycle