Profile cover
H

AI Optical Flow Estimation Engineer

Hannah Cohen

Full-time · Job · Los_Angeles

About the role

Prixel Labs builds foundational computer vision capabilities for autonomous systems and AR/VR applications. We've invested heavily in 3D reconstruction and depth estimation — optical flow is the next capability we're building out, and we need an engineer who can own it end to end. In this role, you'll implement and extend state-of-the-art optical flow estimation models, evaluate performance across diverse scene types (outdoor driving, indoor robotics, AR overlay contexts), and build the evaluation harness that benchmarks our models against RAFT, FlowFormer, and other published baselines. You'll also work with the robotics team to integrate flow estimates into their motion planning pipeline. This is an individual contributor role with a tight research-to-engineering feedback loop. You'll read papers, implement methods, run benchmarks, and ship. We don't have long planning cycles — if a new method looks promising on Monday, we expect a prototype by Friday. Mid-level with clear growth potential on the research track.

Responsibilities

  • Implement and extend state-of-the-art optical flow estimation models
  • Build evaluation harness benchmarking against published baselines (RAFT, FlowFormer)
  • Evaluate model performance across diverse scene types for robotics and AR contexts
  • Integrate flow estimates into downstream motion planning pipelines
  • Document model design decisions and performance trade-offs for internal research log

Requirements

  • 3+ years computer vision ML engineering experience
  • Experience implementing optical flow or dense motion estimation methods
  • Strong PyTorch and Python skills
  • Familiarity with Sintel, KITTI, or similar optical flow benchmark datasets
  • Ability to implement published methods quickly from papers

Benefits

  • Fully remote
  • Research-first culture with paper reading sessions
  • Health and dental insurance
  • GPU compute credits for personal experiments
  • $2,000 annual conference and learning budget

Keep exploring related roles

Use these topic pages to discover more roles in the same hiring cluster.

Categories

AI & Machine Learning Engineers

Skills

python pytorch tensorflow computer vision numpy

Related remote AI jobs

Nearby roles that share similar skills, categories, or hiring intent.

Browse all jobs

Senior AI Researcher- Pre-training (f/m/d)

RemoteAI company

Full-time Senior

PhD Autonomy Engineer Intern - Computer Vision / Deep Learning Summer 2027

RemoteAI company

Full-time Intern

AI Resident UK/EMEA

RemoteAI company

Full-time

Senior AI R&D Engineer – Model Training (f/m/d)

RemoteAI company

Full-time Senior

Senior Machine Learning Engineer (Clinical Team)

RemoteAI company

Full-time Senior

Conversational Modelling Research Engineer

RemoteAI company

Full-time

Job Type

Full-time

Level

Mid-Level

Language

English

Salary Range

$100k – $130k / year

AI Expertise

AI & Machine Learning Engineers

Skills

python pytorch tensorflow computer vision numpy

Ready to apply for this role?

Create a free talent account in under 2 minutes.

  • Apply to verified AI companies
  • Get AI-matched job recommendations
  • Message hiring managers directly
  • Build your public AI talent profile
Create free account & apply Log in

Interested in this role? Create your free profile to apply.

Apply — create free profile