A

ABB

Electrification, automation, and robotics solutions

AI Software Engineer Intern

Applied Machine LearningPosted on Nov 13
InternshipShanghai, ChinaNo salary listed

About the Job

Requirements

  • A bachelor's or master's degree in Computer Science, Software Engineering, Artificial Intelligence, Automation, Electronic Engineering, or a related field is required.
  • Good Python programming skills are required.
  • A basic understanding of machine learning or deep learning, including model training, inference, and evaluation, is required.
  • The candidate must be able to perform structured testing, analyze data, and write clear technical reports in English or Chinese.
  • The candidate must be able to work on-site in Shanghai for at least 3 days per week.

Responsibilities

  • Set up, configure, and maintain edge AI development boards and their software environments.
  • Deploy and run different AI models on NVIDIA Jetson and Huawei Atlas platforms.
  • Design and execute structured tests covering model accuracy, latency, throughput, memory usage, stability, and resource consumption.
  • Compare model and platform performance under consistent test conditions, record issues, and summarize findings.
  • Support data preparation, training, and fine-tuning of small AI models, including experiment tracking and result validation.
  • Prepare clear test plans, test reports, benchmark summaries, and technical documentation.
  • Automate repetitive deployment, test, and data-processing activities using scripts where practical.
  • Collaborate with software and AI engineers to troubleshoot model conversion, runtime, and hardware compatibility issues.

Desired Qualifications

  • Experience with Linux, Git, and shell scripting is preferred.
  • Hands-on experience with PyTorch, TensorFlow, or similar frameworks is preferred.
  • Experience with NVIDIA Jetson, Huawei Ascend/Atlas, CUDA, CANN, ONNX, or model deployment is a strong plus.
  • A commitment of 6 months or longer is preferred.

About the Company

ABB

Electrification, automation, and robotics solutions

ABB provides electrification and automation technologies for industry and infrastructure. Its products include smart building systems, electric vehicle charging, robotics, motors and drives, and factory automation software, delivered as integrated hardware and software that manage energy and automate tasks. ABB differentiates itself with a long history in power and automation, a broad global footprint, and a focused portfolio around electrification, motion, process automation, and robotics, complemented by digital services. Its goal is to help customers use energy more efficiently and work more productively, enabling sustainable and resource-efficient operations and transportation.

Company size
10,001+
Company stage
IPO
Headquarters
Zurich, Switzerland
Founded
2007

Recently Posted Jobs

Explore more opportunities at ABB

See more jobs