Machine Learning Systems Engineer - Remote

Overview

We are seeking a Machine Learning Systems Engineer to help build the world's largest end-to-end 3D native machine learning systems, focusing on developing an ML framework dedicated to 3D.

In Short

• Work within the AI model team to streamline 3D data into high-throughput pipelines.
• Train, accelerate, and deploy machine learning models for 3D GenAI.
• Design and implement reliable and scalable distributed training pipelines.
• Collaborate with researchers, software engineers, and artists.
• Build training infrastructure for in-house foundational models.
• Optimize for high throughput & efficient distributed model training.
• Implement and maintain 3D specific custom operators.
• Build efficient inference endpoints with complex model pipelines.
• Optimize models through compilation and quantization.
• Work in a globally distributed team.

Requirements

• Experience in machine learning or high-performance graphics.
• Understanding of machine learning frameworks like PyTorch.
• Proficient in Python and/or C++ coding.
• Ability to learn quickly and understand complex codebases.
• Detail-oriented with a focus on performance and efficiency.
• Strong communication skills.

Benefits

• Competitive salary, equity, and benefits package.
• Work with a talented team at the forefront of AI and 3D technology.
• Flexible work environment with remote options.
• Opportunities for professional growth and development.
• Inclusive culture valuing creativity and collaboration.