CareerPlan

Machine Learning Engineer – HPC

ShanghaiFull-time2025-09-02 → 2026-11-11

Core

Design, implement, and optimize GPU computing kernels to accelerate training and inference for next-generation 3D generative AI models.

Role type

Senior IC machine learning engineer (HPC/GPU)

Builds

High-efficiency, production-ready GPU modules and domain-specific libraries for 3D generation workloads

Domain

Generative AI, Computer Graphics, High-Performance Computing

Deliverable

production ML models

Required skills

CUDA programming, C++, Python, parallel programming, performance tuning, numerical computation

Preferred skills

model quantization, model compression, computer graphics pipelines, geometry processing, GPU profiling

Technologies

CUDA, Nsight, nvprof