LLM Algorithmic Optimization Engineer - Intern

About the position

NIO is a pioneer and a leading company in the premium smart electric vehicle market. Founded in November 2014, NIO's mission is to shape a joyful lifestyle. NIO aims to build a community starting with smart electric vehicles to share joy and grow together with users. NIO designs, develops, jointly manufactures and sells premium smart electric vehicles, driving innovations in next-generation technologies in autonomous driving, digital technologies, electric powertrains and batteries. NIO differentiates itself through its continuous technological breakthroughs and innovations, such as its industry-leading battery swapping technologies, Battery as a Service, or BaaS, as well as its proprietary autonomous driving technologies and Autonomous Driving as a Service, or ADaaS. NIO's product portfolio consists of the ES8, a six-seater smart electric flagship SUV, the ES7 (or the EL7), a mid-large five-seater smart electric SUV, the ES6, a five-seater all-round smart electric SUV, the EC7, a five-seater smart electric flagship coupe SUV, the EC6, a five-seater smart electric coupe SUV, the ET7, a smart electric flagship sedan, and the ET5, a mid-size smart electric sedan.

Responsibilities

• Conduct research and apply cutting-edge technologies to optimize Large Language Models (LLMs) and multimodal models, exploration and implementation of the core algorithmic optimization on heterogeneous architectures, for highly efficient LLM inference as well as deployment across distributed and heterogeneous hardware environments.
• Focus on model optimization from a systems perspective, ensuring efficient deployment in the vehicle's digital cockpit and advanced driving (AD) domain.
• Collaborate with cross-functional teams to ensure the integration of optimized models into real-world automotive applications.
• Contribute to the entire pipeline from research, development, and testing, through to deployment on hardware, including GPUs and other distributed systems.

Requirements

• Currently pursuing or completed a PhD or Master's degree in Computer Science, Computer Engineering, Applied Mathematics, Communications, Electronics, or a related field with relevant research projects and publications.
• Strong understanding of GPU/NPU architecture and optimization techniques to identify and address bottlenecks.
• Proficient in LLM and VLM architectures and algorithms, familiar with transformer based NLP / Audio / CV algorithms and technologies.
• Proficiency in Python and experience with AI-related training and inference tools such as PyTorch.
• Proficiency in C/C++ programming, familiar with at least one commonly used LLM inference engines.
• Hands-on experience with model-serving frameworks such as Open Neural Network Exchange (ONNX).
• Familiarity with debugging code in distributed computing environments.
• Experience in LLM inference optimization on resource constrained edge devices is a plus.

Nice-to-haves

• Ph.D. in computer science, artificial intelligence, or related fields; or Masters degree + 3 years of relevant industry experience
• Experience in inference optimization techniques of deep learning models or libraries on hardware architectures
• Familiar with microkernel architecture, Linux kernel, hypervisor, middleware, and application framework
• Those who have good publication records and have published high impact, innovative papers are preferred.

Benefits

• CIGNA EPO, HSA, and Kaiser HMO medical plans with $0 for Employee Only Coverage.
• Dental (including orthodontic coverage) and vision plan. Both provide options with a $0 paycheck contribution covering you and your eligible dependents.
• Company Paid HSA (Health Savings Account) Contribution when enrolled in the High Deductible CIGNA medical plan.
• Healthcare and Dependent Care Flexible Spending Accounts (FSA).
• 401(k) with Brokerage Link option.
• Company paid Basic Life, AD&D, short-term and long-term disability insurance.
• Employee Assistance Program.
• Sick and Vacation time.
• 13 Paid Holidays a year.
• Paid Parental Leave for first 8 weeks at full pay (eligible after 90 days of employment with NIO).
• Paid Disability Leave for first 6 weeks at full pay (eligible after 90 days of employment with NIO).
• Voluntary benefits including: Voluntary Life and AD&D options for you, your spouse/domestic partner and dependent child(ren), pet insurance.
• Commuter benefits.
• Mobile Cell Phone Credit.
• Healthjoy mobile benefit app supporting you and your dependents with benefit questions on the go & support with benefit billing questions.
• Free lunch and snacks.
• Onsite gym.
• Employee discounts and perks program.