Third Dimension AI Logo

Third Dimension AI

Research Engineer - Data Infrastructure/ML

Reposted 12 Days Ago
Remote or Hybrid
Hiring Remotely in CA
Mid level
Remote or Hybrid
Hiring Remotely in CA
Mid level
The role involves designing scalable data pipelines for 3D, video, and sensor data, optimizing infrastructure, and productionizing ML models with researchers.
The summary above was generated by AI
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. 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. 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.
In this role, you will:
  • 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.

Preferred Qualifications & Skills
  • Strong programming background in Python.
  • 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).
  • Knowledge of dataset versioning, experiment tracking, and reproducibility tools (DVC, MLflow, etc.).

Nice to Haves
  • Exposure to simulation, robotics, or autonomy testing pipelines.
  • Experience with deep learning frameworks and ML workflows (PyTorch).
  • Familiarity with 3D or robotics data formats (point clouds, meshes, radiance fields, LIDAR).
  • Contributions to open-source ML/infra projects or publications in top-tier AI/ML/CV venues.
  • Experience with frontend / UI work.

Location
  • London, UK / Graz, Austria/ Bay Area: This is a priority for us, as we're trying to build a presence around our already existing teams in London, Graz or the Bay area. Hybrid (remote + in office days) preferred.
  • Remote: If hybrid is not an option, but you're amazing, we're open to having you work out your home location.

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.
  • Hybrid-first culture – State-of-the-art London workspace with the flexibility to work remotely.
  • 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.
  • Community & connection – Team offsites, events, and opportunities to connect with global colleagues.

Similar Jobs

An Hour Ago
Remote or Hybrid
Canada
Senior level
Senior level
Cloud • Insurance • Payments • Software • Business Intelligence • App development • Big Data Analytics
Lead design and rollout of a fit-for-purpose data governance program: manage data lifecycle, implement governance frameworks and tools, drive data catalogs/lineage, upskill stewards, define KPIs, and partner cross-functionally to improve data quality, compliance, and usability.
Top Skills: AIData CatalogsData Governance ToolsData LineageMetadata ManagementSQL
8 Hours Ago
Remote or Hybrid
Senior level
Senior level
Artificial Intelligence • Cloud • HR Tech • Information Technology • Productivity • Software • Automation
Sell SaaS licenses and generate new business through account and territory planning, prospect research, and field sales. Build C-suite relationships, lead cross-functional account strategies, advise on IT roadmaps and AI integration, identify specialists to win deals, negotiate contracts, and meet sales targets. Role requires travel and collaboration with solutions, success, partners, and marketing teams.
Top Skills: AISaaSServicenow
8 Hours Ago
Easy Apply
Remote
Canada
Easy Apply
Senior level
Senior level
Artificial Intelligence • Fintech • Hardware • Information Technology • Sales • Software • Transportation
Lead end-to-end COGS and gross margin analysis across hardware and software product lines. Build scenario models, quantify cost-saving opportunities, and present defensible recommendations to C-suite. Tackle ambiguous executive questions, partner with data engineering, finance, and product to establish reliable datasets, and drive analytical best practices and documentation. Operate autonomously with high ownership.
Top Skills: SQL

What you need to know about the Calgary Tech Scene

Employees can spend up to one-third of their life at work, so choosing the right company is crucial, not just for the job itself but for the company culture as well. While startups often offer dynamic culture and growth opportunities, large corporations provide benefits like career development and networking, especially appealing to recent graduates. Fortunately, Calgary stands out as a hub for both, recognized as one of Startup Genome's Top 100 Emerging Ecosystems, while also playing host to a number of multinational enterprises. In Calgary, job seekers can find a wide range of opportunities.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account