Staff Software Engineer, ML Infrastructure

About the role

Voxel's perception system is the technical core of everything we ship. Our models detect human activity, equipment interactions, environmental hazards, and operational state in real time across thousands of cameras in manufacturing, logistics, retail, and pharmaceutical environments. Safety was our wedge; it proved our platform works. Now customers are pulling us into operations: equipment utilization, workflow compliance, process efficiency. Every new use case runs through the perception team.

Responsibilities

• Set the technical direction for ML infrastructure at Voxel: what we build, what we buy, and how the pieces fit together as the team and model portfolio scale
• Architect and build the training infrastructure that lets the applied ML team run multiple experiments concurrently and iterate quickly on new architectures (PyTorch, AWS)
• Own the train-to-deploy handoff: export trained models to optimized inference formats (TensorRT, ONNX), quantify accuracy and latency impact, and partner with Platform on production deployment
• Pick and roll out the experiment tracking and lifecycle stack (Weights & Biases, MLflow, ClearML, or similar) so researchers can run, compare, and reproduce experiments efficiently
• Establish DevOps-for-ML best practices (IaC, CI/CD, observability, cost monitoring) so researchers can iterate quickly and safely
• Mentor engineers across Vision & AI on ML infrastructure best practices, raising the bar for how the org thinks about training, evaluation, and deployment
• Anticipate where the infrastructure needs to be in 12 to 18 months, including the upcoming move to on-device inference, and architect for that future

Requirements

7+ years building and shipping large-scale software systems, with at least 3 years focused on ML infrastructure or large-scale data infrastructure

A track record of being the person who decides the architecture, not just the person who implements it. You've owned tool selection, framework choices, and build-vs-buy calls for systems other engineers depend on

Deep fluency in PyTorch and the modern ML training stack. You know what good experiment tracking looks like, what makes a training pipeline reliable at scale, and where the failure modes live

Strong Python. Performant, maintainable code that holds up in production

A pragmatic shipping orientation. You can tell the difference between architectural decisions that need to be right and ones that can be revisited later, and you don't over-engineer the latter

Strong communication skills. You can explain complex tradeoffs clearly to ML researchers, infra peers, and leadership

Qualifications

Nice to Have

• Production experience on AWS (S3, EC2, EKS, or similar) for ML workloads
• Hands-on experience with model export and inference optimization (TensorRT, ONNX, or similar), including measuring accuracy and latency tradeoffs against training-time baselines
• Experience with modern ML orchestration tools (Ray, Sematic, Flyte, Metaflow, Prefect, or similar)
• Familiarity with GPU performance profiling and optimization (Nsight, PyTorch profiler, or similar)
• Background in computer vision model training

Skills

Equity through Voxel’s Equity Incentive Plan

Total compensation includes base salary, annual bonus, and equity

Comprehensive health, dental, and vision insurance

Competitive paid parental leave

Unlimited PTO and flexible work arrangements

Daily meals in-office, team events, annual company onsite

Compensation Range: $220K - $260K