Senior Machine Learning Engineer (3D Data)

Join our Foundation AI organization as a Senior Machine Learning Engineer specializing in 3D data. In this role, you will be at the forefront of our foundation model efforts, building the infrastructure that enables our models to perceive and generate virtual worlds in three dimensions. You will collaborate closely with AI researchers, design and implement robust data pipelines, and ensure our foundation models receive the highest quality data. This position requires a strong background in data systems engineering, Python proficiency, and a passion for generative AI.

- Architect and maintain automated pipelines for the ingestion, cleaning, and pre-processing of 3D datasets spanning petabytes of data. - Collaborate closely with ML Engineers and Data Scientists to understand their data requirements, build tooling that streamlines their data workflows, and troubleshoot data-related system issues. - Design and optimize high-throughput, low-latency Inference APIs for internal and external consumer access.

- 5+ years of experience as a research-focused data systems engineer (preferably working with 3D generative models) - Python Proficiency: You can write high-quality Python code for automation, tooling, and infrastructure management - We are seeking accomplished software engineers with a passion for data, experience building large distributed systems, and a commitment to writing high-quality, well-tested code to solve complex data challenges at scale - Team-Player & Technical Leader: You are a collaborative team member who actively mentors peers, drives technical excellence, and takes ownership of leading and delivering key features and projects across team boundaries - A Bachelor's degree or equivalent experience in Computer Science, Computer Engineering, or a similar technical field - Expertise in building scalable ML data pipelines for both batch and real-time environments. Experience working with and processing very large datasets (Petabytes or more) - Are passionate about the potential of generative AI, particularly in creative domains like 3D/4D content - Versatile: You're a generalist and you are comfortable with several languages and technologies already; you are adaptable in any situation - Custom Tooling & Annotation Infrastructure: Proven track record of developing internal human-in-the-loop workflows and annotation tooling tailored to 3D or video data - Game Engines & DCC Tools: Hands-on experience developing in Roblox Studio or working with digital content creation tools such as Blender, Unreal Engine, or Unity - MLOps & Data Versioning Expertise: Experience with experiment tracking platforms (e.g., Weights & Biases, MLflow) and managing versioning for massive-scale datasets - Low-Level Optimization (C++): Strong C++ proficiency to optimize data-loader performance and extend underlying core engine functionality