Stack AV is hiring a
Staff Software Engineer, ML Platform, Remote - United States, Remote
![Logo of Stack AV](https://cdn.jobscollider.com/logo/stack-av-338f.webp)
Staff Software Engineer, ML Platform
🏢 Stack AV
💵 ~$150k-$180k
📍United States, Remote
Please let Stack AV know you found this job on JobsCollider. Thanks! 🙏
Summary
The job is for a Machine Learning Engineer in Stack AV's ML Data team. The role involves building infrastructure for machine learning training and implementing high throughput inference service. The ideal candidate has experience with ML platforms, scalable infrastructure, and working across teams.
Requirements
- Experience with both ML Platforms and building ML-based applications
- Ability to work across teams
- Experience building or using ML infra built for a large number of customer teams
Responsibilities
- Help build infrastructure to support machine learning training
- Implement high throughput inference service using LLMs and vector db
- Set the direction for auto-labeling
Preferred Qualifications
- Deep understanding of design tradeoffs and ability to articulate those tradeoffs and work with others on getting alignment
- Experience with building ML models or ML infra in the domains of autonomous vehicles, perception, and decision making
- Experience with model training, model optimization, or large data processing pipelines
Benefits
- Knows how to push the GPU to its limit from Python to CUDA kernel level (bonus point)
- Built the inference or training loop for a large model (ideally with LLM flavor)
- Shipped ML products (NLP, computer vision, recommender systems, etc.) at scale to make business impact
- Has data platform experience where you built infrastructure for real time querying / vector databases, batch/stream processing using Ray, Spark or similar, and Parquet-based object storage (data lake / data warehouse)
- Knows how to build low latency / high throughput batch or stream processing pipelines
- Knows how to write (readable) high performance C++
- Has prior AV experience
Share this job:
Disclaimer: Please check that the job is real before you apply. Applying might take you to another website that we don't own. Please be aware that any actions taken during the application process are solely your responsibility, and we bear no responsibility for any outcomes.
Similar Jobs
- 💰~$72k-$104k📍India
- 💰~$142k-$196k📍United States, Remote
- 💰~$48k-$59k📍United States, Remote
- 💰~$160k-$280k📍Portugal
- 💰~$155k-$195k📍India
- 💰$191k-$286k📍United States
- 💰$120k-$200k📍India
- 💰~$150k-$190k📍United States
- 💰~$144k-$185k📍Canada
Please let Stack AV know you found this job on JobsCollider. Thanks! 🙏