AI Engineer, Computer Vision

The role

You will train and evaluate visual localization models in PyTorch. You will build data pipelines, run large-scale GPU training experiments, and support production deployment.

What you will do

• Implement, train, and evaluate visual localization models in PyTorch
• Iterate on architectures, loss functions, and training loops
• Contribute to high-throughput ingestion and preprocessing pipelines
• Run training jobs across GPU environments on a 30-billion-image corpus
• Build and maintain the evaluation suite for localization
• Work with senior engineers to take models from checkpoint toward serving
• Bridge research and product by understanding how field failures translate into training data

Required qualifications

• Experience training deep learning models for computer vision tasks in PyTorch
• Foundational understanding of visual localization, learned feature matching, image retrieval, pose estimation, SLAM, or structure-from-motion
• Understanding of 3D geometry including camera models, epipolar geometry, and pose parameterizations
• Comfort working with large image or 3D datasets
• Familiarity with GPU-based training and reasoning about training runs
• Strong Python skills
• Bachelor's or master's degree in computer science, computer vision, robotics, or a related field, or equivalent practical experience

Nice to have

• Published or contributed to work on learned localization, visual place recognition, scene-coordinate regression, or vision foundation models
• 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 (TensorRT, ONNX, CoreML, or similar)
• Experience with petabyte-scale spatial data: 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 CV system

Benefits

• Medical coverage
• Dental coverage
• Vision coverage
• 401(k)
• Comprehensive benefits package

About the company

Niantic Spatial builds spatial AI models and services so people and machines can reason, learn and act safely in the physical world. They turn camera imagery, video, sensor data and archives into accurate, AI-readable representations of real places.