Stability AI is looking for an ML Data Engineer, Research, with a background in 3D computer graphics, to join their Data team. The role aims to improve and scale the efficiency within their models by building the data infrastructure that manages data and powers the training of all Stability AI models, especially 3D generative models.
Requirements
- Background in real-time or offline 3D graphics and rendering (either academic or industry)
- Experience with 3D technologies such as Blender, Houdini, 3ds Max, Maya, and with concepts such as PBR materials and tools for PBR editing (e.g., Substance 3D)
- Experience with shader development (especially for offline rendering)
- Experience with common high-end 3D renderers (e.g., Cycles, VRay, Octane, Renderman, Arnold, Corona)
- Experience with procedural 3D data generation
- Experience with real-time 3D graphics and shader APIs programming (e.g., OpenGL, Vulkan, Direct3D, GLSL, HLSL)
- Experience with image processing and computer vision methods and frameworks (e.g., Open3D, OpenCV)
Responsibilities
- Ingest, clean, normalize, and preprocess data in a scalable, parallelizable way to prepare it for our machine learning model training pipelines while ensuring data quality
- Design, implement, and maintain scalable data infrastructure for generative AI
- Develop tools to search, curate, and serve the data at scale
- Collaborate with research teams to understand and meet their data requirements
- Maintain and improve data quality and integrity across various databases and data stores
- Manage and organize large-scale unstructured data, including image, text, audio, video and 3D
- Keep up to date with the latest data trends and research papers and implement novel ways to create synthetic data
Other
- Good teamwork and communication skills based on experience working with a distributed international team with timezone and cultural differences
- Experience working effectively remotely
- Excellent communication skills to effectively collaborate with users, solve issues, and provide guidance
- Attention to detail and the ability to document processes and solutions effectively
- We are an equal opportunity employer and do not discriminate on the basis of race, religion, national origin, gender, sexual orientation, age, veteran status, disability or other legally protected statuses.
