Company Name: Stability AI

Job Details: Hiring,Remotely,in,United,States,Remote,Mid,level

Job Url: https://builtin.com/job/machine-learning-data-engineer-research-3d-data-focus/7565527

Job Description: (North America remote)ML Data Engineer, Research - 3D Data FocusAbout the role:We are looking for a talented ML Data Engineer, Research, with a background in 3D computer graphics, to join our Data team. You will work alongside a growing multidisciplinary team of talented research scientists and machine learning engineers to improve and scale the efficiency within our models. You will contribute to building the data infrastructure that manages our data and powers the training of all Stability AI models (especially 3D generative models). The position is full-time and 100% remote.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 qualityDesign, implement, and maintain scalable data infrastructure for generative AIDevelop tools to search, curate,  and serve the data at scaleCollaborate with research teams to understand and meet their data requirementsMaintain and improve data quality and integrity across various databases and data storesManage and organize large-scale unstructured data, including image, text, audio, video and 3DKeep up to date with the latest data trends and research papers and implement novel ways to create synthetic dataQualifications: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)Experience in dealing with a diverse range of 3D geometry representations and file formats (e.g., triangular or quad meshes, voxels, NeRFs, SDFs, pointclouds, etc) Experience with a diverse range of 2D image file formats (e..g, 16/32bit image depth, HDR data in formats such as OpenEXR).  Experience with automation and scripting for 3D asset pipelines (e.g., Blender BPY, MEL, HScript)Strong understanding of modern 3D file formats such as USD and glTFProven background within large-scale distributed workloads, and multi-component pipeline synchronization and orchestration Experience fine tuning large generative models (e.g., diffusion, language, or multimodal) on domain-specific datasetsExperience with cloud storage and file systems. AWS (S3) is strongly preferred, but open to other cloud platformsExperience with Python, and additional languages such as C/C++, JavaScriptExpertise in data warehouses and lakehouse technologies (Postgres, MongoDB, BigQuery, Delta Lake, Iceberg, etc ).Experience working with AI research teams and curating data for themGood teamwork and communication skills based on experience working with a distributed international team with timezone and cultural differencesExperience working effectively remotelyExcellent communication skills to effectively collaborate with users, solve issues, and provide guidanceAttention to detail and the ability to document processes and solutions effectivelyEqual Employment Opportunity: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.