Data Infra & ML Engineer - 3D AI Pipelines (Hybrid/Remote)

third-dimension; San Francisco (CA); Hybrid; USD 150,000 - 210,000; Full time

Benefits offered by this job

Competitive salary & stock options

Pension / retirement plan

Health & wellness coverage

Flexible time off

Hybrid-first culture

Workspace support

Learning & development

Community & connection

Job summary

Third Dimension is building SuperSim, a scalable platform for AI-driven 3D data. We seek a Data Infrastructure / ML Engineer to design pipelines for 3D, video, image, and sensor data and to empower researchers to train, validate, and deploy models quickly.

You will work across cloud and containerized environments (Kubernetes, Ray, Dask) and collaborate with researchers to productionize ML models (PyTorch), develop data APIs, and improve data quality and lineage throughout the ML lifecycle.

Qualifications

• Strong programming background in Python.
• Proficiency with distributed data-processing technologies (e.g., Ray, Apache Spark, Flyte, Dask).
• Hands-on experience with cloud infrastructure (AWS, GCP, or Azure), Kubernetes, and distributed training frameworks (Ray, RLLib, PyTorch DDP, or Horovod).
• Knowledge of dataset versioning, experiment tracking, and reproducibility tools (DVC, MLflow, etc.).

Responsibilities

• Build and maintain high-performance data pipelines to ingest, transform, and version multi-modal datasets (3D, video, sensor).
• Design and optimize distributed training and data-processing infrastructure across cloud and containerized environments (Kubernetes, Ray, Dask, EKS).
• Collaborate with researchers to productionize ML models (PyTorch), from prototype to deployment.
• Develop tools and APIs that make data discoverable, reusable, and reproducible.
• Monitor and improve data quality, lineage, and performance across the ML lifecycle.

Skills

Python

Distributed data processing

Cloud infrastructure

Kubernetes

PyTorch

Ray

Dask

Spark

MLflow

DVC

Tools

PyTorch

Ray

Apache Spark

Flyte

Dask

Kubernetes

AWS

GCP

Azure

RLLib

Horovod

Job description

Third Dimension is building SuperSim, a scalable platform for AI-driven 3D data. We seek a Data Infrastructure / ML Engineer to design pipelines for 3D, video, image, and sensor data and to empower researchers to train, validate, and deploy models quickly.

You will work across cloud and containerized environments (Kubernetes, Ray, Dask) and collaborate with researchers to productionize ML models (PyTorch), develop data APIs, and improve data quality and lineage throughout the ML lifecycle.