5 months ago
Shenzhen, ChinaMid Level
Responsibilities
- Design, implement, and optimize GPU computing kernels for 3D GenAI model training and inference.
- Develop and maintain domain-specific libraries and performance-critical components for 3D generation workloads.
- Collaborate with researchers and infrastructure engineers to identify bottlenecks, benchmark performance, and deliver production-ready GPU modules.
Requirements
- Hands-on experience with CUDA and GPU programming.
- Strong programming skills in C++ and Python.
- Solid understanding of parallel programming, performance tuning, and numerical computation.
Benefits
- Experience with quantization, model compression, or other efficiency-oriented model optimization techniques is preferred.
- Knowledge of computer graphics, rendering pipelines, or geometry processing is preferred.
- Familiarity with GPU profiling tools such as Nsight or nvprof, or with hardware-aware optimization, is preferred.
