Meshy is looking to build the world's largest 3D GenAI systems and needs Machine Learning Engineers to help streamline 3D data into high-throughput pipelines and scale training infrastructure.
Requirements
- Solid knowledge of at least one machine learning framework (e.g. PyTorch, Tensorflow).
- Very strong programming skills and ability to write clean and maintainable research code in Python and/or C++.
- A strong passion to navigate through the PyTorch internals, with hands-on experience in areas like GPU efficiency optimization, data loader optimization, mixed precision training and JIT compilation.
- Experiences with large-scale distributed training, familiarity with modern parallelization techniques: DDP, MP, PP, FSDP, etc.
- Experience with diffusion models in 3D or video.
Responsibilities
- Work within the AI model team to streamline 3D data into high-throughput pipelines and scale training infrastructure to hundreds of GPUs.
- Train, accelerate, and deploy machine learning models for 3D GenAI.
- Design and implement reliable and scalable distributed training pipelines, optimize end-to-end training efficiency.
- Work closely with researchers, software engineers, and artists to integrate AI models into production.
Other
- 3+ years of relevant engineering or research experience in machine learning, computer vision and/or graphics.
- Excellent problem-solving skills and the ability to work independently and as part of a team.
- Strong communication skills and the ability to convey complex technical concepts to a non-technical audience.
- Competitive salary, equity, and benefits package.
- Flexible work environment, with options for remote and on-site work.
