Staff Software Engineer - Machine Learning
$145k - $284k • Remote - United States • Full-time
Posted 12d ago
Remote Work Policy
On-site
Employment Type
Full-time
Categories
Machine Learning Engineer
About the job
The Smart Agents group is responsible for building ML models and systems to simulate road users in various situations, generating scenarios for testing and training AV driving policies. This involves developing ML/AI models that control other characters in a simulation environment to interact realistically with autonomous vehicles. The technology stack includes Generative AI models (GPT) and Reinforcement Learning (RL) policies. The team collaborates closely with Simulation, Behaviors, Perception, and Safety Engineers. Specific duties include ML/RL model development, training loop development, optimization, integration, creating ML infrastructure, metrics, and data pipelines for production deployment and rapid experimentation.
Responsibilities
- Develop machine learning (ML) and reinforcement learning (RL) models, including training loop development and optimization.
- Streamline integration and create ML infrastructure, metrics, and data pipelines for production model deployment and rapid experimentation.
- Contribute strong software engineering (SWE) expertise to the ML team.
- Accelerate project timelines, particularly in areas related to Autopilot, Lane Keep, and autonomous vehicle (AV) technologies.
- Deploy perception/prediction/AV models into real-world environments.
- Work with RL and sequence prediction (ML) models.
Requirements
- 4+ years of experience in robotics or latency-sensitive backend services.
- Background working with machine learning teams, algorithms, and models.
- Experience building highly performant ML and system pipelines.
- Strong programming skills in modern C++ or Python.
- Experience with profiling CPU and/or GPU software, process scheduling, and prioritization.
- Expertise in setting scalable, efficient, fault-tolerant, and extensible architectures.
- Ability to design across multiple systems and understand areas outside of your domain.
- Ability to shift between coding, design, technical strategy, and mentorship.
- Proven experience in machine learning and classification.
- Familiarity with ML frameworks such as Tensorflow or PyTorch.
Benefits
- Bonus Potential: An incentive pay program offers payouts based on company performance, job level, and individual performance.
- Health and wellbeing benefit programs including medical, dental, vision, Health Savings Account, Flexible Spending Accounts.
- Retirement savings plan.
- Sickness and accident benefits.
- Life insurance.
- Paid vacation & holidays.
- Tuition assistance programs.
- Employee assistance program.
- GM vehicle discounts.