Senior Machine Learning Engineer

We are seeking Machine Learning Engineers to help build the world's largest 3D GenAI systems. The ideal candidate will have strong engineering skills, extensive industry experience, and excel in a fast-paced, high-ownership environment.

Responsibilities:

• Develop high-throughput pipelines for 3D data and scale training infrastructure to hundreds of GPUs.
• Train, accelerate, and deploy 3D GenAI models.
• Collaborate with researchers, engineers, and artists to integrate AI models into production.
• Optimize and scale models for performance and efficiency.

Requirements:

• 3+ years of experience in machine learning, computer vision, or graphics.
• Proficiency in at least one ML framework (e.g., PyTorch, TensorFlow).
• Strong programming skills in Python and/or C++.
• Excellent problem-solving abilities and teamwork skills.
• Strong communication skills and ability to explain technical concepts to non-technical audiences.
• Familiarity with cloud platforms (AWS, Google Cloud, Azure).

Preferred:

• Experience with 3D data and computer graphics.
• Hands-on experience with GPU programming (CUDA, torch.compile, JAX, XLA).
• Knowledge of generative models (GANs, VAEs, diffusion models).
• Experience with software development best practices (Git, testing, CI/CD).