AI Engineer, Computer Vision

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.