Lead Ai Infrastructure Engineer
- Countries
- USA
- Job Types
- software & tech, engineering & hardware
- Experience
- late
- Organization Types
- transportation & mobility, robotics, startups
- Location
- Austin
- Posted Date
- Company
- Avride
AI Summary
Job Description
Lead the ML infrastructure layer, focusing on optimizing GPU inference for onboard and offboard applications. Responsibilities include taking ownership of the GPU inference framework, improving performance, and collaborating with the applied ML team on neural model architecture.
Hard Requirements
- Experience with PyTorch
- Understanding of how GPUs work
- Experience in diagnosing and resolving performance issues
- Strong record of building infrastructure including distributed systems
- 5+ years of experience with C++
- Programming experience in multi-threaded environments
Soft Requirements
- Leadership mindset