Design and train modern computer vision models (CNNs, Vision Transformers, foundation models) to solve novel perception tasks
Build and maintain scalable training pipelines using PyTorch and HPC infrastructure (e.g., Slurm, distributed training)
Develop data curation and active learning workflows
Optimize models for deployment (ONNX, TensorRT, containerization)
Test and validate models in both cloud and edge environments
Build reproducible experimentation workflows (version control, experiment tracking, configuration management)
Drive experimental cycles: define hypotheses, implement techniques from literature, evaluate results, and present recommendations
Translate research ideas into production-ready implementations
Design, implement, and test ML-related components and supporting software
Perform statistical analysis and fine-tune model performance based on production and field feedback

Requirements

3+ years of full-time professional experience in computer vision-focused machine learning (not student internship)
Strong PyTorch experience, including custom layers, loss functions, datasets, dataloaders, and training loops
Experience with modern vision architectures (CNNs, ViTs, DETR-style models, foundation models)
Experience building or contributing to production ML systems
Solid software engineering fundamentals (testing, version control, clean code principles)
Strong communication skills and ability to explain complex technical topics clearly
Engineering degree or equivalent practical experience

Benefits

Pre-IPO stock options (tax-advantaged ISOs)
Competitive base salary
Medical, dental, and vision insurance - 100% of premiums paid for employees and 85% of premiums paid for dependents
Generous paid time off and holidays
401(k) Plan
An inclusive and tight company culture that is mission driven

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

computer visionCNNsVision TransformersPyTorchHPC infrastructureONNXTensorRTstatistical analysisversion controlexperiment tracking

Soft Skills

strong communication skillsability to explain complex technical topicsproblem-solvingcollaborationcritical thinkingadaptabilityattention to detailtime managementpresentation skillshypothesis definition

Certifications

engineering degreeequivalent practical experience