Machine Learning System Engineer

About Meshy

Headquartered in the Silicon Valley, Meshy is the leading 3D generative AI company on a mission to Unleash 3D Creativity. Meshy makes it effortless for both professional artists and hobbyists to create unique 3D assets—turning text and images into stunning 3D models in just minutes. What once took weeks and $1,000 now takes 2 minutes and $1.

Our global team of top experts in computer graphics, AI, and art includes alumni from MIT, Stanford, Berkeley, as well as veterans from Nvidia and Microsoft. With 3 million users (and growing), Meshy is trusted by top developers and backed by premiere venture capital firms like Sequoia and GGV.

• No. 1 popularity, among 3D AI tools, according to A16Z games,
• No. 1 website traffic, among 3D AI tools, according to SimilarWeb (2M monthly visits),
• Leading 3D foundation model, delighted texture & fine geometry,
• $52M funding by Top VCs,
• 2.5M users & 20M models generated!

Ethan Yuanming Hu serves as the founder and CEO. Ethan got his Ph.D. in graphics and AI from MIT, where he developed the Taichi GPU programming language (27K stars on GitHub, used by 300+ institutes). His Ph.D. thesis got a honorable mention of SIGGRAPH 2022 Outstanding Doctoral Dissertation Award and his research has been cited over 2700 times. His favorite animal is the llama.

About The Role

We are looking for Machine Learning Systems Engineers who can help us build the world's largest end-to-end 3D native machine learning systems. You will help us build our end to end ML framework dedicated for 3D, from pretraining, to finetuning, inferencing, etc. We expect a combination of strong hands on engineering skills, eagerness to learn new things, and thrives in a fast-paced, high-ownership environment.

What You’ll Do

• 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.

On the training side

• Work closely with researchers to build the training infrastructure for our in-house foundational models.
• Identifying bottlenecks and optimizing for high throughput & efficient distributed model training across hundreds to thousands of GPUs.
• Building and maintaining training clusters and job schedulers.
• Implementing and maintaining 3D specific custom operators in Triton or CUDA

On the inference side

• Building efficient inference endpoints with complex model pipelines
• Optimizing models through compilation, fusion, quantization, etc.

What We're Looking For

• Experience in machine learning or high performance graphics.
• Solid practical understanding of at least one machine learning framework (e.g. PyTorch, Flax).
• Strong ability to write beautiful and maintainable code in Python and/or C++.
• Ability to learn fast and dive into new concepts or complex codebases.
• Performance and efficiency oriented mindset, with a strong interest in the tiniest detail.
• Strong communication skills for working in a globally distributed team.

Nice To Have

• A strong passion to navigate through the PyTorch internals, with hands-on experience in areas like torch.compile , fully_shard (FSDP2) APIs.
• Experience with building Triton kernels.
• Experiences with large-scale distributed training, familiarity with modern parallelization techniques: DP, TP, CP, PP, zero redundancy optimizers, etc.
• Experience with diffusion models in 3D or video.
• Experience with full bf16 or partially fp8 training.

Our Values

• Brain: We value intelligence and the pursuit of knowledge. Our team is composed of some of the brightest minds in the industry.
• Heart: We care deeply about our work, our users, and each other. Empathy and passion drive us forward.
• Gut: We trust our instincts and are not afraid to take bold risks. Innovation requires courage.
• Taste: We have a keen eye for quality and aesthetics. Our products are not just functional but also beautiful.

Why Join Meshy?

• Competitive salary, equity, and benefits package.
• Opportunity to work with a talented and passionate team at the forefront of AI and 3D technology.
• Flexible work environment, with options for remote and on-site work.
• Opportunities for fast professional growth and development.
• An inclusive culture that values creativity, innovation, and collaboration.
• Unlimited, flexible time off.

Benefits

• Competitive salary, benefits and stock options.
• 401(k) plan for employees.
• Comprehensive health, dental, and vision insurance.
• The latest and best office equipment.