Staff Machine Learning Engineer
XPeng Motors
🇺🇸 United States
On-site
Staff / Principal
3 weeks ago
$179,400 – $303,600
- AI
- Machine Learning
- Computer Vision
- PyTorch
- TensorFlow
- Python
- ONNX
- TensorRT
- Equity
3 weeks ago
- Research and develop algorithms for reinforcement learning based methods for planning.
- Design efficient model architectures that can run in real-time on the computing platform of our vehicles.
- Develop offline data-driven ML infrastructure for fast adaptation of the planning ML models.
- Deliver on target planning SW and closely work with the perception team to achieve the most intelligent autonomous driving systems.
- Work with massive field-testing data to continuously improve autonomous driving technologies.
- Designing, running, and analyzing experiments and testing to evaluate the efficiency of our solutions on real-world data.
- Partnering with system software engineering specialists to ship industrial strength ML models.
- Communicating and collaborating with multi-functional teams.
- Education in Engineering, Robotics, Computer Science with a focus on Reinforcement Learning, Artificial Intelligence, or a related field, or equivalent experience.
- Strong experience in applied reinforcement learning including model architecture design, model training, data mining, and data analytics.
- 3-5 + years of experience working with machine learning frameworks such as PyTorch, Tensorflow.
- Strong Python programming experience with software design skills.
- Solid understanding of data structures, algorithms, code optimization and large-scale data processing.
- Excellent problem-solving skills.
- MS or PhD level education in Engineering, Robotics or Computer Science with a focus on Reinforcement Learning, Artificial Intelligence, or a related field, or equivalent experience.
- Related experience in autonomous driving or robotics production solutions.
- Hands on experience in developing RL based planning engine for autonomous driving or robotic system.
- Experience in model deployment and optimization tools such as ONNX and TensorRT.
- A fun, supportive and engaging environment.
- Infrastructures and computational resources to support your work.
- Opportunity to work on cutting edge technologies with the top talents in the field.
- Opportunity to make a significant impact on the transportation revolution by the means of advancing autonomous driving.
- Competitive compensation package.
- Snacks, lunches, dinners, and fun activities.
Staff Machine Learning Engineer · XPeng Motors