Senior Software Engineer, ML Infrastructure

Software EngineerVoxel'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.We're hiring a strong software engineer to own the ML Infrastructure that powers how Voxel trains and ships vision models. You'll build systems that let our applied ML team train multiple models concurrently, manage experiments and ship optimized models to production. You'll set technical direction, write code, make architecture calls, and partner closely with applied CV, ML Data and Platform engineers.What You'll DoBuild and maintain training infrastructure that lets the applied ML team train multiple models concurrently, manage experiments, and iterate quickly on new architectures.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.Establish ML experiment tracking and lifecycle management - pick the right tools (Weights & Biases, MLflow, ClearML, or similar) so researchers can run, compare, and reproduce experiments efficiently.Establish DevOps-for-ML best practices on AWS (IaC, CI/CD, observability, cost monitoring) so researchers can iterate quickly and safely.Understand the infra needs of applied ML/CV engineers and design scalable solutions that support model development.What We're Looking For4+ years of experience building and shipping large scale software solutions.Hands-on experience building ML training pipelines in PyTorch.Hands-on experience with ML experiment tracking and lifecycle tools (Weights & Biases, MLflow, ClearML, or similar).Experience with AWS (S3, EC2, EKS, or similar) for ML workloads.Strong Python. Write performant code that scales well in production environments.Track record of owning infrastructure end-to-end: scoping, building, shipping, and improving systems that internal teams depend on.Bias toward shipping. You'd rather ship something good this week than something perfect next quarter.Strong communication skills.Nice to HaveExperience 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 trainingCompensation & BenefitsEquity through Voxel's Equity Incentive PlanTotal compensation includes base salary, annual bonus, and equityComprehensive health, dental, and vision insuranceCompetitive paid parental leaveUnlimited PTO and flexible work arrangementsDaily meals in-office, team events, annual company onsite