Staff Software Engineer - ML Infrastructure

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 seeking a highly skilled and experienced Senior ML Infrastructure professional to join their team. This role will focus on designing and implementing systems to support Voxel's ML development. The ideal candidate will have a strong background in software engineering, extensive experience working with distributed systems, and a deep understanding of infrastructure and ML systems. What You'll Do • Build and maintain cloud infra and distributed systems for MLOps • Build out internal training framework to make research with large distributed models easy and fun • Design and develop systems to support Voxel's ML development, with a focus on computer vision applications. • Provide technical guidance, mentorship, and project management support. • Build distributed systems and pipelines for data management, ensuring scalability, reliability, and performance. • Demonstrate a deep understanding of DevOps practices and apply them to ML operations. • Utilize technologies such as Kubernetes and various CI/CD systems to optimize infrastructure deployment and management. • Handle the complexities of designing and implementing ML infrastructure systems in a dynamic and fast-paced startup environment. • Contribute to the overall impact of Voxel's AI-powered workplace safety solutions by delivering robust and efficient ML infrastructure. Must Haves • Bachelor's degree in Computer Science or a related field. • Minimum 5+ years of experience in software engineering, with a focus on infrastructure design. • Have experience designing large, highly available distributed systems with Kubernetes. • Proven experience in designing complex systems and strong software engineering skills. • Strong understanding and experience with distributed systems like Apache Spark, Ray etc. and infrastructure design. • Proficiency in containerization technologies like Docker and orchestration platforms like Kubernetes. • Demonstrated expertise in DevOps practices, with a focus on ML. • Have worked on ML tools that researchers love Nice-to-Haves: • Knowledge of advanced ML operations techniques, such as model deployment and monitoring. • Experience with pipeline automation and data management in ML workflows.Previous experience building ML systems and working on ML adjacent teams. 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