Third Dimension AI Logo

Third Dimension AI

Research Engineer - Data Infrastructure/ML

Reposted One Month 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
CA, USA
136K-245K Annually
Senior level
136K-245K Annually
Senior level
eCommerce • Fintech • Hardware • Payments • Software • Financial Services
Own Square Hardware sales and revenue across major U.S. retail partners. Build and scale retail partnership strategies, manage existing retailer and distributor relationships, source and negotiate new partnerships, lead cross-functional launches and forecasting, and manage retail marketing budgets through test-and-learn investments.
2 Hours Ago
Remote or Hybrid
67K-105K Annually
Junior
67K-105K Annually
Junior
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Sells CrowdStrike’s cybersecurity platform to commercial customers in Manitoba and Saskatchewan. Owns the full sales cycle from prospecting through closing, generates net-new business, manages quota, forecasts in Clari, and collaborates with sales development, engineering, channel, and marketing teams. The role requires consultative selling to mid-market and enterprise organizations, executive-level engagement, partner strategy, competitive positioning, and use of sales and AI tools. Occasional travel and modified hours may be required.
Top Skills: Ai TechnologiesClariCloudCybersecurity SolutionsGongLinkedin Sales NavigatorSaaSSalesforce (Sfdc)Zoominfo
6 Hours Ago
Remote or Hybrid
Senior level
Senior level
AdTech • Cloud • Digital Media • Information Technology • News + Entertainment • App development
Lead UX and UI design for player-focused game interfaces. Create wireframes, player flows, interactive layouts, visual direction, style guides, and mockups using Figma. Collaborate with designers, engineers, and artists to align interfaces with gameplay and narrative goals. Support onboarding and mentorship, while ensuring usability, accessibility, responsiveness, and intuitive experiences. Help establish visual language standards across products and contribute to UI implementation for PC and console platforms.
Top Skills: BlueprintsFigmaMaterial EditorUmgUnreal Engine 5

What you need to know about the Austin Tech Scene

Austin has a diverse and thriving tech ecosystem thanks to home-grown companies like Dell and major campuses for IBM, AMD and Apple. The state’s flagship university, the University of Texas at Austin, is known for its engineering school, and the city is known for its annual South by Southwest tech and media conference. Austin’s tech scene spans many verticals, but it’s particularly known for hardware, including semiconductors, as well as AI, biotechnology and cloud computing. And its food and music scene, low taxes and favorable climate has made the city a destination for tech workers from across the country.

Key Facts About Austin Tech

  • Number of Tech Workers: 180,500; 13.7% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Dell, IBM, AMD, Apple, Alphabet
  • Key Industries: Artificial intelligence, hardware, cloud computing, software, healthtech
  • Funding Landscape: $4.5 billion in VC funding in 2024 (Pitchbook)
  • Notable Investors: Live Oak Ventures, Austin Ventures, Hinge Capital, Gigafund, KdT Ventures, Next Coast Ventures, Silverton Partners
  • Research Centers and Universities: University of Texas, Southwestern University, Texas State University, Center for Complex Quantum Systems, Oden Institute for Computational Engineering and Sciences, Texas Advanced Computing Center

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account