Third Dimension
Research Engineer - Data Infrastructure/ML
San Francisco, USA
Sponsorship not specifiedDetected 278 days ago
AWSGCPAzureCloud PlatformsKubernetesMachine LearningDeep LearningPyTorchSparkData EngineeringComputer VisionRoboticsSensorsResearch
About the role
- Monitor and improve data quality, lineage, and performance across the ML lifecycle.
- Nice to Haves Exposure to simulation, robotics, or autonomy testing pipelines.
Responsibilities
- Third Dimension is building SuperSim, a new kind of simulator which can enable fast, cost-effective and photorealistic 3D simulations directly from source data using AI.
- You'll design scalable pipelines for 3D, video, image, and other sensor data (e.g., LIDAR) and develop the ML workflows that power SuperSim.
- Your work will enable our researchers to train, validate, and deploy models faster - ultimately shaping how robots and autonomous systems are tested in the real world.
- 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.
Requirements
- Proficiency in 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).
- Experience with frontend / UI work.
- Knowledge of dataset versioning, experiment tracking, and reproducibility tools (DVC, MLflow, etc.).
- Familiarity with 3D or robotics data formats (point clouds, meshes, radiance fields, LIDAR).
Nice to have
- & Skills Strong programming background in Python.
Benefits
- Benefits Competitive salary & stock options - Everyone is an owner and shares in our success.
- Pension / retirement plan - Company contributions to support your long-term financial wellbeing.
- Health & wellness - Comprehensive health, dental, and vision insurance (with regional equivalents for employees outside the UK).
- Flexible time off - Generous holiday allowance plus local public holidays.
- Workspace support - Latest hardware and home office equipment to set you up for success.
- Learning & development - Budget and time for conferences, courses, and workshops to keep you at the frontier of ML and 3D AI.
Company info
- We are working with customers across multiple industries, including autonomous vehicles, drones, robots in industrial and manufacturing environments, and more.
- About the Role We're looking for a Data Infrastructure / ML Engineer to build the backbone of our 3D AI systems.
Apply directly at Third Dimension →Create a free account for alerts like thisView Third Dimension immigration profile
This listing is sourced directly from Third Dimension's careers page and normalized into a canonical job model.