Machine Learning Engineer
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 (GPU optimization)
Builds
High-efficiency, production-ready GPU modules for 3D generation workloads
Domain
Generative AI, Computer Graphics, GPU 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
Responsibilities
Design and optimize GPU kernels for model training and inference; Develop and maintain domain-specific libraries for 3D generation; Collaborate with researchers and infra engineers to benchmark performance and identify bottlenecks
Seniority
Senior, hands-on IC
