AI Engineer, Computer Vision

Niantic; San Francisco (CA); Hybrid; USD 166,000 - 221,000; Full time

Job summary

Niantic Spatial is hiring an AI Engineer, Computer Vision, to contribute to deep learning models for their localization system. Youll train and iterate visual models, working with researchers and engineers to improve feature extractors, matchers, and pose regressors across a full training-to-production loop.

Youll gain exposure to data pipelines, evaluation, and deployment, with emphasis on correctness and scalable experimentation.

Qualifications

• Experience training deep learning models for computer vision tasks using PyTorch.
• Solid understanding of visual localization, image retrieval, pose estimation, and 3D geometry.
• Experience handling large image/3D datasets and data preprocessing.
• Familiarity with GPU-based training and analyzing training runs to identify bottlenecks.
• Strong Python programming skills and clean, reproducible code.
• Bachelor's or master's degree in CS/CV/Robotics or equivalent practical experience.

Responsibilities

• Train, evaluate, and iterate visual localization models in PyTorch.
• Build data pipelines for imagery, poses, and 3D point clouds.
• Run scalable experiments across GPU clusters and ensure reproducibility.
• Extend the evaluation suite for pose accuracy and latency.
• Support production deployment from checkpoint to serving.
• Bridge research and product to translate field failures into improvements.

Skills

PyTorch

Python

Deep learning

Computer vision

GPU training

Data preprocessing

Education

CS/CV/Robotics degree

Tools

CUDA

TensorRT

ONNX

Job description

About Niantic Spatial

At Niantic Spatial, we're building the future of physical AI. Powered by a proprietary database of over 30 billion posed images, our groundbreaking mapping technology unlocks a new dimension of interaction and spatial intelligence that helps both humans and machines better understand, represent, navigate, and engage with the real environment.

Our reconstruction technology captures environments with geometric accuracy and extreme detail from any standard camera, and our Visual Positioning System delivers precise positioning almost anywhere in the world. We serve customers across robotics, the public sector, and energy and industrial markets — building for the 80% of economic activity that takes place beyond our screens.

About the Team

Localization is where the map meets the moment. Our localization team builds the living geospatial world model and the centimeter-level Visual Positioning System (VPS) that lets people and robots know exactly where they are, from any camera, almost anywhere on Earth.

That system is learned, not hand-built. Our localization stack has moved from classical geometry to visual models trained on billions of posed images — models that have to be accurate to the centimeter, robust to lighting and seasons and clutter, and fast enough to run in production for every query. This team sits at the junction of research and production: we take what works in a paper, make it work on our data, and make it work for customers.

About the Role

We're hiring an AI Engineer, Computer Vision, to contribute to the deep learning models at the heart of our localization system. Working alongside experienced researchers and engineers, you'll help train and iterate on the visual models — feature extractors, matchers, pose regressors, and retrieval models — that turn a single image into a precise position in the world.

This is a full-loop environment. You'll get exposure to the entire pipeline: the data the model learns on, training, evaluation, and the service that runs it in production. You'll be surrounded by people who care deeply about correctness — where a loss curve going down is not the same as a system getting better — and you'll grow into that rigor quickly.

What You'll Do

• Train and Evaluate Models — Implement, train, and evaluate visual localization models in PyTorch: 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.
• Build Data Pipelines — 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 Experiments at Scale — Run training jobs across GPU environments on our 30-billion-image corpus, working with senior engineers and ML Infrastructure to keep experiments reproducible and results interpretable.
• Extend the Benchmark — Help build and maintain the evaluation suite for localization — pose accuracy, recall across lighting and season changes, and latency under production load — so results are comparable across model versions.
• Support Production Deployment — Work with senior engineers to take models from checkpoint toward serving: export, inference optimization, and monitoring once live. Learn what it means to own a model end-to-end, not just hand off a checkpoint.
• Bridge Research and Product — Work with the Research team and Real-World Test Lab to understand how field failures translate into training data and benchmark improvements.

What You'll Bring

• 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, SLAM, or structure-from-motion, and the 3D geometry underneath them (camera models, epipolar geometry, pose parameterizations).
• Comfort working with large image or 3D datasets — 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.

Nice to Have

• Published or contributed to work on learned localization, visual place recognition, scene-coordinate regression, or vision foundation models (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 (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.

Compensation & Benefits

The expected salary range for this role is $165,600 - $221,000 per year. Compensation also includes an annual bonus, equity, and a comprehensive benefits package including medical, dental, and vision coverage, 401(k), and more.

Location & Work Model

This role is based in our San Francisco, CA office on a hybrid schedule with a minimum of 3 days per week in office.

Inclusive Application

We know the strongest candidates don't always tick every box. If you're excited about this role and believe you could do it well, we encourage you to apply even if your experience doesn't match every qualification listed — you may be exactly who we're looking for.

Equal Opportunity

Niantic Spatial is an equal opportunity employer. Individuals seeking employment at Niantic Spatial are considered without regard to race, color, ancestry, national origin, religion, creed, age, gender (including pregnancy, childbirth, breastfeeding or related medical conditions), marital status, physical or mental disability, medical condition, genetic information, military or veteran status, gender identity, gender expression, sexual orientation, or any other protected category under applicable laws. Niantic Spatial will also consider qualified applicants with criminal histories in accordance with applicable laws. Please contact your recruiter if you want to request an accommodation for the job application or interview process.

Candidate Privacy

I understand that by submitting my job application, the information I provide as part of that application will be used in accordance with Niantic Spatial's Privacy Notice for Job Applicants and Candidates https://www.nianticspatial.com/applicant-privacy-notice.