Meshy
Machine Learning Engineer
Shenzhen, Remote. Posted 7 months ago
The role
The Machine Learning Engineer will design, implement, and optimize GPU computing kernels for 3D GenAI models. They will develop domain-specific libraries and work with researchers and infra engineers to deliver high-efficiency GPU modules.
What you will do
- Design, implement, and optimize GPU computing kernels to accelerate model training and inference for next-generation 3D GenAI models.
- Develop and maintain domain-specific libraries and performance-critical components for 3D generation workloads.
- Work closely with researchers and infra engineers to identify bottlenecks, benchmark performance, and deliver high-efficiency, production-ready GPU modules.
Required qualifications
- 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.
Nice to have
- Experience with quantization, model compression, or other efficiency-oriented model optimization techniques.
- Knowledge of computer graphics, rendering pipelines, or geometry processing.
- Familiarity with GPU profiling tools (e.g., Nsight, nvprof) or hardware-aware optimization.
About the company
Meshy is a market leader in 3D generative AI used in game and film production, 3D printing, and industrial design. The company builds a full pipeline for 3D content and a community for creators.