Machine Learning Engineer - LLM, AI & Robotics

XPeng Motors is known for its electric cars and the leading autonomous driving technology. As a technology driven company, we are also developing advanced humanoid robots for various applications in industries. And AI is the core in all the above efforts. With this mission, we are looking for passionated engineers of all levels who will build large language models / foundation models together with us to shape this new future of Robotics & AI. You will work on a lean team without boundaries. Your influence will be on every single product that is shipped, without boundaries as well.

Job Responsibilities

• Fine-tune the pre-trained LLM on particular use cases in Humanoid Robots and Autonomous Driving Cars, e.g. chat, API calling, etc.

• Work on efficient LLMs (e.g. small LLMs, weight sharing, model quantization, etc.) that can be deployed locally in Robots and Cars.

• Develop and maintain the LLM (including multi-modal LLM) training and deployment pipelines that could scale to massive production.

• Work on efficient and effective ways to leverage large scale dataset and computational resource.

• Explore novel ways to generate synthetic data to improve the data diversity & quality.

• Communicate and align with different teams on product needs across the company

Minimum Skill Requirements:

• Ph.D in Computer Science or related fields.

• Solid background in machine learning, natural language processing (NLP), computer vision,

  speech or data.

• Experience in working with LLMs

• Strong python programming skills & software development, and strong experience with any

  major deep learning framework.

•  Open-minded, collaborative and excellent communication skills

Preferred Skill Requirements:

• Deep understanding of language model modern architectures, ‘under the hood’ LLM training knowledge, quantization techniques, etc.

• Familiarity of large language models (LLMs) relevant technologies, such as SFT, RLHF, prompt engineering, RAG, etc.

• Record of publications, innovations or leadership.

What do we provide:

• A fun, supportive and engaging environment

• Opportunity to make significant impact on transportation revolution by the means of

  advancing autonomous driving

• Opportunity to work on cutting edge technologies with the top talent in the field

• Competitive compensation package

• Snacks, lunches, and fun activities

The base salary range for this full-time position is $180,000-$300,000, in addition to bonus, equity and benefits. Our salary ranges are determined by role, level, and location. The range displayed on each job posting reflects the minimum and maximum target for new hire salaries for the position across all US locations. Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant education or training.

We are an Equal Opportunity Employer. It is our policy to provide equal employment opportunities to all qualified persons without regard to race, age, color, sex, sexual orientation, religion, national origin, disability, veteran status or marital status or any other prescribed category set forth in federal or state regulations.