Machine Learning System Engineer

Reposted 19 Days Ago
2 Locations
In-Office or Remote
200K-200K Annually
Mid level
3D Printing • Artificial Intelligence • Information Technology • Internet of Things
The Role
Machine Learning Systems Engineers will develop end-to-end 3D machine learning systems, optimize training pipelines, and improve model performance within a collaborative team.
Summary Generated by Built In
 About Meshy

Headquartered in Silicon Valley, Meshy is the leading 3D generative AI company on a mission to Unleash 3D Creativity by transforming the content creation pipeline. 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 cost $1,000 now takes just 2 minutes and $1.

Our world-class team of top experts in computer graphics, AI, and art includes alumni from MIT, Stanford, and Berkeley, as well as veterans from Nvidia and Microsoft. Our talent spans the globe, with team members distributed across North America, Asia, and Oceania, fostering a diverse and innovative multi-regional culture focused on solving global 3D challenges. Meshy is trusted by top developers, backed by premiere venture capital firms like Sequoia and GGV, and has successfully raised $52 Million in funding.

Meshy is the market leader, recognized as the No.1 in popularity among 3D AI tools (according to 2024 A16Z Games) and No.1 in website traffic (according to SimilarWeb, with 3 Million monthly visits). The platform boasts over 5 Million users and has generated 40 Million models.

Founder and CEO Yuanming (Ethan) Hu earned his Ph.D. in graphics and AI from MIT, where he developed the acclaimed Taichi GPU programming language (27K stars on GitHub, used by 300+ institutes). His work is highly influential, including an honorable mention for the SIGGRAPH 2022 Outstanding Doctoral Dissertation Award and over 2,700 research citations.

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.
  • Located in or near one of our employee hubs — Bay Area, CA; Seattle, WA; New York, NY (NJ); Vancouver or Toronto, Canada.
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
  • Stock options available for core team members.
  • 401(k) plan for employees.
  • Comprehensive health, dental, and vision insurance.
  • The latest and best office equipment.

Top Skills

C++
Cuda
Flax
Python
PyTorch
Triton
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
San Jose, California
27 Employees

What We Do

Meshy empowers artists, game developers, and creators to bring their visions to life with a toolkit for creating 3D models in seconds

Similar Jobs

In-Office or Remote
2 Locations
4820 Employees
101K-168K Annually

Atlassian Logo Atlassian

Principal Machine Learning System Engineer

Cloud • Information Technology • Productivity • Security • Software • App development • Automation
In-Office or Remote
San Francisco, CA, USA
11000 Employees
190K-306K Annually

Wipfli Logo Wipfli

Senior Devops Engineer

Cloud • Fintech • Software • Business Intelligence • Consulting • Financial Services
Remote or Hybrid
United States
3000 Employees
88K-118K Annually

ServiceNow Logo ServiceNow

Senior Talent Sourcer

Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Remote or Hybrid
Kirkland, WA, USA
28000 Employees
103K-169K Annually

Similar Companies Hiring

Credal.ai Thumbnail
Software • Security • Productivity • Machine Learning • Artificial Intelligence
Brooklyn, NY
Standard Template Labs Thumbnail
Software • Information Technology • Artificial Intelligence
New York, NY
10 Employees
Scotch Thumbnail
Software • Retail • Payments • Fintech • eCommerce • Artificial Intelligence • Analytics
US
25 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account