Voxel Logo

Voxel

Staff Software Engineer - ML Infrastructure

Reposted Yesterday
Be an Early Applicant
Hybrid
8 Locations
Senior level
Hybrid
8 Locations
Senior level
Responsible for designing and implementing ML infrastructure systems, building cloud infrastructure, ensuring scalability, and providing mentorship within a fast-paced startup environment.
The summary above was generated by AI
Who Are We
Industrial labor is incredibly dangerous work - almost 3 million people in the US per year are injured in the workplace for entirely preventable and at times, fatal or debilitating causes. Protecting these essential people who power our world is what motivates Voxelitos, and we'd love for you to join us. At Voxel, we're passionate about revolutionizing workplace safety and operations with groundbreaking, full-stack AI and computer vision technology.

Voxel’s site intelligence platform helps safety and operations leaders see the unseen risks, make strategic decisions, and prevent workplace incidents before they happen. Our customers include Fortune 500 companies across major grocers and retailers, manufacturers, food and beverage warehousers, supply chain and logistics service providers. Based in SF with team members sitting all over the globe, Voxel is backed by industry leading VC’s.



Voxel is looking for a Staff Machine-Learning Infrastructure Engineer to drive the next wave of our computer-vision platform for workplace safety. You will be the technical owner for three pillars of our ML lifecycle — ground-truth data & labeling workflows, large-scale training infrastructure, and continuous model lifecycle management. If you excel at designing cloud-native, distributed systems that turn raw video into production-ready, version-controlled models, we’d love to meet you.

What You'll Do

  • Own data & labeling pipelines – architect scalable labeling services (storage, query, retrieval), design ontologies, automate annotation workflows, and build quality-tiered datasets that stay within cost constraints.
  • Build and operate training infrastructure – create multi-GPU / multi-node training frameworks (Ray, Spark, Kubernetes), optimize distributed jobs, and integrate accelerators (TensorRT, CUDA-graph, FP8, etc.).
  • Manage the full model lifecycle – stand up model registries, version control, evaluation suites, and continuous-learning loops that push updates from dev → staging → prod with zero-downtime rollbacks.
  • Provide technical leadership, mentorship, and lightweight project management to a small infra + research squad.
  • Establish DevOps-for-ML best practices (IaC, CI/CD, observability, cost monitoring) so researchers can iterate quickly and safely.
  • Partner with ML engineers on architecture decisions, from data schemas to inference optimizations, ensuring infra and research road-maps stay tightly aligned.

Qualifications (Must Haves)

  • Bachelor’s (or higher) in Computer Science, EE, or related field.
  • 5+ years building and operating large-scale infrastructure, with at least 3 years focused on ML or data-intensive systems.
  • Proven record designing highly available, distributed systems on Kubernetes (EKS, GKE, or on-prem).
  • Deep expertise with orchestration (K8s operators, Argo, Kubeflow), and cluster-scale storage / compute (S3, GCS, Ray, Spark, Dask).
  • Hands-on experience automating data-labeling or ground-truth workflows and maintaining dataset versioning.
  • Strong software-engineering fundamentals; familiar with best practices for testing, observability, and secure coding.
  • Demonstrated DevOps mindset — IaC (Terraform/CDK), CI/CD (GitHub Actions, ArgoCD), metrics & alerting (Prometheus/Grafana).

Nice-to-Haves

  • Experience running multi-instance / multi-GPU training jobs, mixed-precision optimizations, or TensorRT / Triton inference.
  • Familiarity with active-learning, continuous-training, or online distillation pipelines.
  • Background in model registry tooling (MLflow, BentoML, SageMaker Registry) and evaluation dashboards.
  • Prior work with computer-vision models (YOLO, DETR, Faster RCNN) or video understanding at scale.
  • Contributions to open-source ML infra projects or published talks/blogs on MLOps.
  • Exposure to edge-deployment or real-time inference systems.
  • Experience shipping high quality production code in Python

Why Join Us?
Join a visionary team revolutionizing safety and operations, directly impacting the well-being of millions of essential workers. This is your chance to build an extraordinary business and foster a vibrant company culture that demands your absolute best. Alongside AI experts, experienced entrepreneurs, and passionate problem-solvers, you'll play a pivotal role in shaping the company's growth trajectory and market position. Enjoy a competitive salary, benefits, and a dynamic work environment.

Benefits:
Extensive / Generous health, dental, and vision insurance.
Highly competitive paid parental leave and support system.
Ownership in the business through an Equity Incentive Plan.
Generous paid time off and / or flexible work arrangements.
Daily meals in-office, vibrant company events, team-building.
401K retirement plan, HSA options, pre-tax Commuter Card.

Top Skills

Spark
Ci/Cd
Docker
Kubernetes
Ray

Similar Jobs

10 Hours Ago
Hybrid
Vancouver, BC, CAN
Senior level
Senior level
Artificial Intelligence • Hardware • Information Technology • Security • Software • Cybersecurity • Big Data Analytics
The Senior Software Developer will lead technical efforts in C++ and C# for developing security applications, focusing on product design, implementation, and team collaboration.
Top Skills: AzureC#C++Ci/CdDatabasesDevOpsTest Automation
10 Hours Ago
Hybrid
Toronto, ON, CAN
Mid level
Mid level
Artificial Intelligence • Hardware • Information Technology • Security • Software • Cybersecurity • Big Data Analytics
As a Senior Frontend Engineer, you'll develop scalable SPAs for audio/video redaction, collaborating with teams and optimizing user experience.
Top Skills: AWSAzureCSSGCPHTML5JavaScriptKubernetesReactTypescript
10 Hours Ago
Hybrid
Vancouver, BC, CAN
Senior level
Senior level
Artificial Intelligence • Hardware • Information Technology • Security • Software • Cybersecurity • Big Data Analytics
The Senior Software Developer will design, develop, and maintain secure, networked software components for Linux-based video appliances, mentoring team members and collaborating with cross-functional teams.
Top Skills: AIC++Ci/CdDockerGithub ActionsGoIotLinuxOpensslPython

What you need to know about the Vancouver Tech Scene

Raincouver, Vancity, The Big Smoke — Vancouver is known by many names, and in recent years, it has gained a reputation as a growing hub for both tech and sustainability. Renowned for its natural beauty, the city has become a magnet for professionals eager to create environmental solutions, and with an emphasis on clean technology, renewable energy and environmental innovation, it's attracted companies across various industries, all working toward a shared goal: advancing clean technology.

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account