Logo
NIO

Sr/Staff AI Infrastructure Engineer

NIO, San Jose, CA, United States


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.

Roles and Responsibilities:

  • Architect technical solutions that help us deliver AIOS foundation capability in SkyOS across ECU domains.
  • Lead the proof-of-concept development to evaluate the architectural designs in terms of functionality, performance, security, and dependability.
  • Partner with engineering teams to understand real-world constraints and to support the high-quality implementation of AIOS designs for product development, validation and integration.
  • Define the evolution roadmap for AIOS solutions.
  • Define and deliver requirements for development team.
  • Follow industry trends and study competitors’ technical stacks.

Must Qualifications:

  • 3+ years of experience in software design and development for AI model training, and/or inference optimization, and deployment.
  • In-depth knowledge and hands-on experience in the latest AI model architectures used in the industry and research areas, e.g. Transformer.
  • Experience building at least one public AI extension/application/project
  • Experience with cloud infrastructure and training (Azure, AWS, etc.)
  • CI/CD experience (Gitlab, GitHub, Jenkins etc)
  • 3+ years experience working in ML infra such as PyTorch / TensorFlow / etc.
  • 7+ years development experience of building complex software systems.
  • Proficient C/C++ programming and Python.
  • BS / MS in Electrical Engineering, Computer Engineering, Computer Science or equivalent.
  • Excellent communication skills, and technical writing proficiency.
  • Organized mindset, and passionate about delivering high-quality software.

Preferred Qualifications:

  • Masters degree + 3 years of relevant industry experience, or Ph.D. degree in computer science, artificial intelligence, or related fields.
  • Experience building ComfyUI extension or Python native apps using SDXL
  • Understanding of hardware acceleration(Stable Diffusion, SAM, Mistral, Groq, etc)
  • Familiar with LLM and optimization on resource constrained edge devices for inferences
  • Experience in performance tuning & inference optimization techniques of deep learning models or libraries on hardware architectures.
  • In-depth knowledge of computer architectures, CPU / MPU, GPU / NPU.
  • Solid understanding of embedded / mobile operating systems.
  • Familiar with microkernel architecture, Linux kernel, hypervisor, middleware, and application framework.

Compensation:

  • The US base salary range for this full-time position is $195,000 - $220,000.
  • Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant education or training.
  • Please note that the compensation details listed in US role postings reflect the base salary only, and do not include discretionary bonus, equity, or benefits.

Benefits:

Along with competitive pay, as a full-time NIO employee, you are eligible for the following benefits on the first day you join NIO:

  • CIGNA EPO, HSA, and Kaiser HMO medical plans with a $0 paycheck contribution 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