Senior/Staff Software Engineer - Prediction Integration
Zoox
Software Engineering
Foster City, CA, USA
USD 221k-319k / year + Equity
Posted 6+ months ago
The Active Prediction Integration team is responsible for the development of a unified prediction and planning architecture, by integrating ML-based and first-principles-based models. It is a highly cross-functional team that collaborates with planning, prediction and behavior ML, and perception engineers to critically influence robot behaviors. You’ll be part of a passionate team dedicated to enabling Zoox’s rider-first experience by making the autonomous robot safer, smarter and more comfortable.
Together with the team, you will advance the state of the art in prediction and decision-making. We are looking for engineers with experience in developing and integrating advanced ML models in safety-critical production environments. If you enjoy learning new AI technologies and embrace the challenges of evolving prediction and planner architectures, we would like to hear from you.
Responsibilities
- You will work closely with Prediction and Planner to integrate advanced ML models and develop first-principles based safeguards, to ensure trustworthy prediction and decision making.
- You will inform and guide the development of new ML models and should be willing to dive deep and address issues in the model and data pipelines.
- You will design interfaces between prediction and planning, creating abstraction layers that facilitate modular integration of various machine learned components within the planning architecture.
- You will create principled algorithmic improvements to specific areas of driving behavior and implement those changes in our code base.
- You will facilitate the collaborations between the prediction data/modeling teams and the planning and control teams in order to deliver architectural changes to the system.
Qualifications
- Master's or PhD degree in computer science or related field and 7+ years of relevant experience
- Fluency in modern C++ standards, programming and software design principles
- Strong knowledge of algorithmic design, implementation and optimization
- Prior experience or expertise in machine learning, either in model architecture or production data pipelines
- Demonstrated ability to communicate and collaborate cross-functionally
Bonus Qualifications
- Demonstrated ability to create real-time motion planning algorithms
- Experience with sequential decision making, including, but not limited to, imitation learning, reinforcement learning, joint trajectory prediction and planning, etc.
- Experience with machine learning and deep learning frameworks (e.g., TensorFlow, PyTorch, TensorRT) and basic knowledge of CUDA programming or willingness to learn
About Zoox
Zoox is developing the first ground-up, fully autonomous vehicle fleet and the supporting ecosystem required to bring this technology to market. Sitting at the intersection of robotics, machine learning, and design, Zoox aims to provide the next generation of mobility-as-a-service in urban environments. We’re looking for top talent that shares our passion and wants to be part of a fast-moving and highly execution-oriented team.
Accommodations
If you need an accommodation to participate in the application or interview process please reach out to accommodations@zoox.com or your assigned recruiter.
A Final Note:
You do not need to match every listed expectation to apply for this position. Here at Zoox, we know that diverse perspectives foster the innovation we need to be successful, and we are committed to building a team that encompasses a variety of backgrounds, experiences, and skills.