Lead AI / ML Staff Engineer

Reposted 17 Days Ago
San Francisco, CA, USA
In-Office
Mid level
Artificial Intelligence • Software • Design • Generative AI
The Role
Lead development of spatial deep learning models (diffusion and 3D generative) from data curation through training, evaluation, and production deployment; create scalable training datasets from LiDAR, images, videos, and floorplans; combine geometry algorithms with learned models and integrate outputs into an interactive 3D product stack.
Summary Generated by Built In

We are looking for a Lead AI & Algorithms Engineer to own the full lifecycle of novel spatial model development; defining the right framing of complex inference problems, structuring image, video, and vectorized floorplan data into usable training formats, training and evaluating models, and deploying performant systems into our production stack. You’ll be responsible for developing and advancing our diffusion models and 3D generative models that power Dream’s core spatial intelligence.

Upcoming Challenges

  • Create training datasets from imperfect sources (LiDAR Scans, scraped construction docs)

  • Combining heuristics, labeling tools, and clever preprocessing (e.g. object detection models) to generate high-quality data at scale.

  • Design and train novel spatial deep learning models that intelligently infer room layouts, surface boundaries, and object placement from partial inputs — balancing architectural plausibility with flexibility and precision.

  • Explore hybrid approaches that combine classical geometry algorithms with learned models.

  • Benchmark and iterate on model performance across multiple axes — accuracy, speed, generalization

  • Integrate AI models directly into the product stack, working closely with full-stack engineers to ensure outputs are usable in the interactive 3D design environment.

What we Require

  • 4+ Years of experience training and deploying real-world AI . Deep understanding of model architectures, training pipelines, and evaluation methodologies.

  • Creativity and technical depth to explore, evaluate, and iterate on different algorithmic approaches.

  • Scrappiness and speed in turning unstructured or incomplete data into usable training sets, and in getting models built and tested under real constraints.

  • Strong background in algorithms and data structures, with the ability to design performant solutions to spatial and generative problems.

  • Hands-on experience with procedural generation techniques, especially in the context of 3D geometry, modeling, or simulation.

  • Expert proficiency in Python, with a strong preference for experience using static typing (e.g., with type hints, MyPy, or Pydantic).

Skills Required

  • 4+ years training and deploying real-world AI systems
  • Deep understanding of model architectures, training pipelines, and evaluation methodologies
  • Experience designing and training diffusion models and 3D generative models
  • Experience creating training datasets from LiDAR, images, video, and vectorized floorplans
  • Strong background in algorithms and data structures for spatial and generative problems
  • Hands-on experience with procedural generation techniques for 3D geometry, modeling, or simulation
  • Experience combining classical geometry algorithms with learned models
  • Experience integrating AI models into production stacks and working with full-stack engineers
  • Expert proficiency in Python
  • Experience with static typing (type hints, MyPy, Pydantic)
  • Experience using object detection models, labeling tools, and preprocessing to generate training data
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
10 Employees

What We Do

Dream Technologies is a software company focused on redefining the future of home design. The company leverages cutting-edge technology, including Generative AI, Machine Learning, and 3D graphics systems, to create innovative design tools. Their mission is to modernize the home design experience through high-performance engineering and artificial intelligence, as demonstrated by their recruitment of specialized engineers in generative AI and 3D systems.

Similar Jobs

CoreWeave Logo CoreWeave

Senior Manager, Technical Accounting

Cloud • Information Technology • Machine Learning
In-Office
4 Locations
1450 Employees
149K-198K Annually

Samsara Logo Samsara

Integration Engineer

Artificial Intelligence • Cloud • Computer Vision • Hardware • Internet of Things • Software
Easy Apply
Remote or Hybrid
United States
4000 Employees
106K-160K Annually

Tapestry - Coach and Kate Spade Logo Tapestry - Coach and Kate Spade

Store Manager

eCommerce • Fashion • Retail • Sales • Wearables • Design
Remote or Hybrid
14 Locations
16000 Employees
62K-94K Annually

Airwallex Logo Airwallex

Associate Director, Strategy & Operations - CEO Office

Artificial Intelligence • Fintech • Payments • Business Intelligence • Financial Services • Generative AI
Hybrid
San Francisco, CA, USA
2300 Employees
170K-300K Annually

Similar Companies Hiring

Kepler  Thumbnail
Fintech • Software
New York, New York
6 Employees
LTX Thumbnail
Robotics • Conversational AI • Generative AI
Jerusalem, Israel
300 Employees
Onshore Thumbnail
Artificial Intelligence • Fintech • Software • Financial Services
New York, New York
60 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account