N

Niantic Spatial

Develops large-scale geospatial model for AR

AI Engineer - Computer Vision

Computer VisionUpdated on Nov 11
Full-timeSan Francisco, CA, USA

About the Job

Requirements

  • Experience training deep learning models for computer vision tasks in PyTorch through coursework, research, internships, or early industry work.
  • Foundational understanding of at least one of visual localization, learned feature matching, image retrieval, pose estimation, simultaneous localization and mapping (SLAM), or structure-from-motion, and the underlying 3D geometry, including camera models, epipolar geometry, and pose parameterizations.
  • Comfort working with large image or 3D datasets, including preprocessing, curation, and understanding how data quality affects model behavior.
  • Familiarity with GPU-based training and the ability to read and reason about training runs, identifying whether a bottleneck is in data loading, compute, or model architecture.
  • Strong Python skills and a habit of writing code that others can rerun and build on.
  • A bachelor's or master's degree in computer science, computer vision, robotics, or a related field, or equivalent practical experience.

Responsibilities

  • Implement, train, and evaluate visual localization models in PyTorch, including local feature extraction and matching, image retrieval, pose regression, and scene-coordinate regression; iterate on architectures, loss functions, and training loops to improve accuracy on real production captures.
  • Contribute to high-throughput ingestion and preprocessing pipelines that turn raw imagery, poses, and 3D point clouds into training-ready datasets, including curation, sampling, and augmentation.
  • Run training jobs across GPU environments on the 30-billion-image corpus, working with senior engineers and ML Infrastructure to keep experiments reproducible and results interpretable.
  • Help build and maintain the localization evaluation suite for pose accuracy, recall across lighting and season changes, and latency under production load, so results are comparable across model versions.
  • Work with senior engineers to take models from checkpoint toward serving, including export, inference optimization, and monitoring once live.
  • Work with the Research team and Real-World Test Lab to understand how field failures translate into training data and benchmark improvements.

Desired Qualifications

  • Published or contributed to work on learned localization, visual place recognition, scene-coordinate regression, or vision foundation models, such as CVPR, ECCV, ICCV, NeurIPS, 3DV, or equivalent.
  • Experience with large-scale self-supervised or multi-view pretraining, or fine-tuning vision foundation models for geometric tasks.
  • Exposure to inference optimization for edge or mobile hardware using TensorRT, ONNX, Core ML, or similar tools.
  • Experience with petabyte-scale spatial data, including point clouds, meshes, posed image collections, or geospatial indexing.
  • Background in robotics, augmented reality, 3D reconstruction, digital twins, or autonomous systems.
  • Prior industry or research internship experience shipping or contributing to a production computer vision system.

About the Company

Niantic Spatial

Develops large-scale geospatial model for AR

Niantic Spatial builds a Large Geospatial Model that uses a proprietary database of over 30 billion posed images to give machines and people a spatially grounded understanding of the real world. Its platform relies on a third-generation digital map with high fidelity to capture world content and enable new AR experiences. The product combines large-scale machine learning with geospatial data to connect scenes globally, supporting developers and businesses that want to add spatial intelligence to their apps through licensing and partnerships. Niantic Spatial differentiates itself by focusing on a globally connected, semantically rich spatial model and a strong developer ecosystem around AR and geospatial applications. The company aims to help customers build more capable spatial apps by providing advanced spatial reasoning, licensing its technology and collaborating through partnerships and events.

Company size
51-200
Company stage
Growth Equity (Venture Capital)
Total funding
$550M
Headquarters
San Francisco, CA
Founded
2021

Recently Posted Jobs

Explore more opportunities at Niantic Spatial

See more jobs