Machine Learning Data Engineer, Research with 3D Data Focus

stability ai • United State
Remote
Apply
AI Summary

We are looking for a talented ML Data Engineer, Research, to join our Data team. This role involves improving and scaling the efficiency of our models using 3D computer graphics and related technologies. The ideal candidate will have experience with 3D data generation and management, as well as large-scale distributed workloads and multi-component pipeline synchronization.

Key Highlights
Improve and scale the efficiency of Stability AI models
Contribute to building data infrastructure for 3D generative models
Develop tools to search, curate, and serve 3D data at scale
Key Responsibilities
Ingest, clean, normalize, and preprocess 3D data in a scalable, parallelizable way
Design, implement, and maintain scalable data infrastructure for generative AI
Collaborate with research teams to understand and meet their data requirements
Technical Skills Required
Blender Houdini 3ds Max Maya PBR materials Substance 3D SDL GLSL HLSL OpenGL Vulkan Direct3D OpenCV
Benefits & Perks
100% remote work
Full-time employment status
Nice to Have
Experience with shader development and 3D rendering

Job Description


(North America or Canada remote)

ML Data Engineer, Research - 3D Data Focus

About 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 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

Qualifications:

  • 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 glTF
  • Proven 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 datasets
  • Experience with cloud storage and file systems. AWS (S3) is strongly preferred, but open to other cloud platforms
  • Experience with Python, and additional languages such as C/C++, JavaScript
  • Expertise in data warehouses and lakehouse technologies (Postgres, MongoDB, BigQuery, Delta Lake, Iceberg, etc )
  • Experience working with AI research teams and curating data for them
  • 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

Equal 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.

Similar Jobs

Explore other opportunities that match your interests

Data Analyst (Remote)

Data Science
•
58m ago
Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Mid-Senior level

Lensa

United State

Data Analyst

Data Science
•
9h ago
Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Associate

Wiraa

United State

Senior Clinical Data Scientist

Data Science
•
9h ago
Visa Sponsorship Relocation Remote
Job Type Contract
Experience Level Mid-Senior level

The Fountain Group

United State

Subscribe our newsletter

New Things Will Always Update Regularly