(Senior) Software Engineer, Deep Learning

Pony.ai; Fremont (CA); On-site; USD 140,000 - 280,000; Full time

Benefits offered by this job

Retirement Plan

Life Insurance

Paid Time Off

Family Leave

Short Term & Long Term Disability

Free Food & Snacks

Job summary

Pony.ai is seeking a Machine Learning Engineer in Fremont, California, to develop and optimize deep learning models for self-driving vehicles. You’ll work with a team on software architecture, design, and deployment of cutting-edge ML approaches that include reinforcement learning and large language models.

The ideal candidate will have a Master’s degree in Computer Science or equivalent industry experience, and strong skills in C++ and Python. The position offers a competitive salary range of $140,000 to $280,000 annually, along with a comprehensive benefits package including a retirement plan and paid time off.

Qualifications

• Master’s degree in Computer Science or 2+ years of relevant industry experience.
• Solid understanding of data structures, algorithms, and large-scale data processing.
• Experience in applied machine learning including data collection and analysis.
• Strong communication skills and team spirit.

Responsibilities

• Work with experts on software architecture and design for self-driving vehicles.
• Design and develop large-scale foundation models using real-world data.
• Develop and deploy deep learning models including vision language models.
• Optimize models to run robustly under tight run-time constraints.

Skills

Machine Learning

Deep Learning

C++

Python

Data Structures

Algorithms

Education

Master in Computer Science or equivalent

PhD in related field (preferred)

Tools

CUDA

TensorRT

Job description

Responsibility

• Work with experts in the field of self-driving vehicles on software architecture and design, system and module design, evaluation metrics, specification and implementation of test and regression frameworks.
• Design and develop large-scale foundation models trained on vast real world data.
• Frame the open-ended real-world problems into well-defined ML problems; develop and apply cutting-edge ML approaches (deep learning, reinforcement learning, imitation learning, etc.) to these problems; scale them to data pipelines; and streamline them to run in real-time on the cars.
• Develop and deploy deep learning models, including vision language models (VLMs) and Large Language Models (LLMs).
• Optimize deep learning models to run robustly under tight run-time constraints.

Requirements

• Master in Computer Science, or at least 2 years of equivalent industry experience in similar technical fields.
• Solid understanding of data structures, algorithms, parallel computing, code optimization and large scale data processing.
• Experience in applied machine learning including data collection and analysis, evaluation and feature engineering.
• Expertise in C++/Python.
• Strong communication skills and team spirit.

Preferred Experience

• PhD in Deep Learning, Machine Learning, Robotics, Natural Language Processing, or similar technical field of study.
• Publications on top-tier conferences like CVPR/ICCV/ECCV/ICLR/ICML/NeurIPS/ICLR/AAAI/IJCV/PAMI.
• Experience in applying ML/DL for behavior prediction, imitation learning, motion planning.
• Experience in deploying deep learning algorithms for real time applications, with limited computing resources.
• Experience in convex optimization, computational geometry or linear algebra.
• Experience in GPU/CUDA/TensorRT.

Compensation and Benefits

Base Salary Range: $140,000 - $280,000 Annually

Compensation may vary outside of this range depending on many factors, including the candidate’s qualifications, skills, competencies, experience, and location. Base pay is one part of the Total Compensation and this role may be eligible for bonuses/incentives and restricted stock units.

Also, we provide the following benefits to the eligible employees:

• Retirement Plan (Traditional and Roth 401k)
• Life Insurance (Basic, Voluntary & AD&D)
• Paid Time Off (Vacation & Public Holidays)
• Family Leave (Maternity, Paternity)
• Short Term & Long Term Disability
• Free Food & Snacks

Please click here for our privacy disclosure.