Title:
Senior Software Engineer - ML/Data Infrastructure
About UnitX:
UnitX is building the world’s best robotics product to accelerate human productivity in manufacturing. UnitX is a fast-moving startup with a team from Stanford and Google. Since inception, UnitX has shipped 1000+ mission-critical systems across 100+ of the world's leading manufacturers' production lines. Every year, $2.8B dollar worth of products (think EV batteries) go through UnitX AI inspection system to ensure quality.
If you are a customer-obsessed, first-principle problem solver who loves getting into the technical nitty gritty to deliver compelling, quantifiable, and scalable value to our customer within the 24/7 mission-critical manufacturing world, there is truly no better place to do it than UnitX. We have assembled a world-class team of problem solvers in Silicon Valley and would love to talk to you.
Role Overview:
We’re looking for a Senior Software Engineer – ML/Data Infrastructure to lead the design and development of scalable systems that support our ML workflows and infrastructure, including data collection, annotation, model training, deployment, and observability. You will own critical backend components and collaborate closely with ML engineers, 2d/3d vision engineers, and application developers to create a rock-solid foundation for production AI.
Sample projects:
Introduce TensorRT into our edge inference system to speed up inference performance and repeatability
Optimize Generative AI pipeline reliability and scalability on our cloud-based Generative AI product
Integrate open-source AI data cleaning pipeline with our internal data and ML experiment system
What You’ll Do:
Design and build scalable pipelines for data ingestion, preprocessing, and annotation from 2D/3D vision systems
Develop infrastructure for distributed model training, versioning, and experimentation tracking
Build robust APIs and tools for ML model deployment, monitoring, and rollback
Integrate with robotic and edge devices to support real-time ML inference in production
Ensure data quality, reproducibility, and system observability across the ML lifecycle
Collaborate across teams to define and implement best practices for ML Ops and Data Engineering
Mentor junior engineers and contribute to the evolution of a high-performance engineering culture
Our Tech Stack:
Technical Domains: Deep learning, Computer vision, 3D imaging, Hardware, Edge computing, Cloud computing, Realtime systems, Industrial systems
Frontend: React, Vue 3, TypeScript
Backend: Python, Java, PostgreSQL, gRPC, PyTorch, OpenCV
What We're Looking For:
Bachelor's degree in Computer Science, Computer Engineering, or a relevant technical field
5+ years of experience building backend or infra systems in data- or ML-intensive environments
Deep understanding of distributed systems, data engineering, and ML lifecycle
Proficiency with Python and backend architecture for data/ML systems
Experience with edge deployment, hardware integration, or real-time systems
Experience with containerization, orchestration, and cloud-native deployments
Strong grasp of ML Ops best practices and experience supporting teams of ML practitioners
Ownership mentality, excellent communication, and startup-ready agility
Benefits:
Competitive salary & equity
Unlimited PTO
Full Medical, Dental, Vision, 401k
Daily meals provided with your own choice
If an employer mentions a salary or salary range on their job, we display it as an "Employer Estimate". If a job has no salary data, Rise displays an estimate if available.
Are you ready to take your expertise to the next level? At UnitX, we're searching for an adventurous Senior Software Engineer specializing in ML/Data Infrastructure in sunny Santa Clara. Our mission? To build cutting-edge robotics technology that boosts productivity in manufacturing. Working with a world-class team drawn from premier institutions like Stanford and Google, you'll play a critical role in advancing our performance metrics, ensuring that our 1000+ systems deliver quality assurance for over $2.8 billion worth of products each year. In this position, you'll lead the design and development of scalable systems that support our machine learning workflows, including the ins and outs of data collection, annotation, and model training. You’ll collaborate with talented ML engineers and application developers to forge a strong foundation for our cutting-edge AI applications. Projects like optimizing the Generative AI pipeline or integrating open-source solutions will keep you engaged and challenged. With a tech stack that includes Python, Java, PyTorch, and cloud computing technologies, there's ample opportunity for innovation and mentorship. If you're passionate about setting industry standards while working in a dynamic, startup environment, join us at UnitX and help redefine manufacturing for the better!
UnitX is seeking a proactive Product Manager for their AI and Imaging Software team to lead innovative product development efforts.
Join UnitX as a Product Manager specialized in 3D Robotics, driving innovative automation solutions in a rapidly evolving environment.
Join Nutanix’s Frontline team as a Software Engineer, where you will develop innovative cloud products using Java, Spring Boot, and React in a supportive and collaborative environment.
Esper is looking for a Backend Engineer to elevate data-driven policymaking in government through innovative software solutions.
As a key player in our startup, you'll develop a cutting-edge simulation platform that empowers AI agents to perform software engineering tasks.
An exciting opportunity for an experienced SIP Developer to advance VoIP technologies in a flexible remote role.
As a Senior Engineer at Cardinal Health, you will lead SAP solutions for the Pharma business while working remotely.
Join Palo Alto Networks as a Sr Staff Engineer Software, where you will shape the future of cybersecurity through innovative cloud management solutions.
Join Net2phone as an AI Senior Golang Developer to lead the development of groundbreaking AI solutions while collaborating with talented professionals.
Join Lucanet as a Senior Backend Engineer and be a key player in transforming financial analysis through innovative software solutions.
Subscribe to Rise newsletter