Data Infra & ML Engineer for 3D AI Simulations

Thirddimension; Greater London; Hybrid; GBP 90,000 - 130,000; Full time

Benefits offered by this job

Competitive salary & stock options

Pension / retirement plan

Health & wellness

Flexible time off

Hybrid-first culture

Workspace support

Learning & development

Community & connection

Job summary

Third Dimension is seeking a Data Infrastructure / ML Engineer to build the backbone of our 3D AI systems in London or remotely. You’ll design scalable pipelines for 3D, video, image, and sensor data and develop the ML workflows that power SuperSim.

You’ll collaborate with researchers to productionize PyTorch models, create tools to make data discoverable and reproducible, and monitor data quality across the ML lifecycle. We offer a hybrid-first culture and stock options.

Qualifications

• Strong programming background in Python.
• Proficiency in distributed data-processing technologies (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, Buildkite).
• 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

Distributed training

Dataset versioning

Experiment tracking

Reproducibility tools

Tools

Ray

Apache Spark

Flyte

Dask

DVC

MLflow

PyTorch

Job description

Third Dimension is seeking a Data Infrastructure / ML Engineer to build the backbone of our 3D AI systems in London or remotely. You’ll design scalable pipelines for 3D, video, image, and sensor data and develop the ML workflows that power SuperSim.

You’ll collaborate with researchers to productionize PyTorch models, create tools to make data discoverable and reproducible, and monitor data quality across the ML lifecycle. We offer a hybrid-first culture and stock options.