Let’s get started
By clicking ‘Next’, I agree to the Terms of Service
and Privacy Policy, and consent to receive emails from Rise
Jobs / Job page
Senior Software Engineer - ML/Data Infrastructure image - Rise Careers
Job details

Senior Software Engineer - ML/Data Infrastructure

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

Average salary estimate

$140000 / YEARLY (est.)
min
max
$120000K
$160000K

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.

What You Should Know About Senior Software Engineer - ML/Data Infrastructure, UnitX

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!

Frequently Asked Questions (FAQs) for Senior Software Engineer - ML/Data Infrastructure Role at UnitX
What are the responsibilities of a Senior Software Engineer - ML/Data Infrastructure at UnitX?

At UnitX, the Senior Software Engineer - ML/Data Infrastructure is responsible for designing and building scalable pipelines for data ingestion, preprocessing, and annotation from 2D/3D vision systems. Collaborating closely with ML engineers, you will develop infrastructure for distributed model training and implement APIs for ML model deployment, ensuring system observability across the entire machine learning lifecycle. Your role also includes mentoring junior engineers and defining best practices for ML Ops within the high-performance engineering culture at UnitX.

Join Rise to see the full answer
What qualifications and experience are required for the Senior Software Engineer - ML/Data Infrastructure position at UnitX?

To be considered for the Senior Software Engineer - ML/Data Infrastructure role at UnitX, candidates should have a Bachelor's degree in Computer Science, Computer Engineering, or a related technical field. Ideally, you should bring at least 5 years of experience building backend or infrastructure systems in data- or ML-intensive environments. A deep understanding of distributed systems, proficiency in Python, and familiarity with cloud-native deployments and ML Ops best practices are essential for success in this position.

Join Rise to see the full answer
How does UnitX support continuous learning and development for the Senior Software Engineer - ML/Data Infrastructure?

At UnitX, we foster an environment of continuous learning and development for all team members, including our Senior Software Engineer - ML/Data Infrastructure. Mentorship opportunities abound as you collaborate with talented engineers and cross-functional teams; we encourage you to share knowledge and best practices in ML Ops and data engineering. Additionally, we provide resources for personal and professional development, ensuring you have the support you need to excel and grow in your career.

Join Rise to see the full answer
What technologies will the Senior Software Engineer - ML/Data Infrastructure work with at UnitX?

As a Senior Software Engineer - ML/Data Infrastructure at UnitX, you will be working with a diverse tech stack that includes deep learning frameworks such as PyTorch and software languages like Python and Java. Your role will utilize cloud computing technologies and handle real-time systems, while also engaging with tools for data management and APIs. Familiarity with containerization and orchestration tools will be beneficial as you integrate with robotic and edge devices.

Join Rise to see the full answer
What is the company culture like at UnitX for Senior Software Engineer - ML/Data Infrastructure?

The company culture at UnitX is fast-paced, innovative, and focused on solving complex problems. As a Senior Software Engineer - ML/Data Infrastructure, you can expect a collaborative environment with a team that values your input and encourages agile thinking. You’ll find yourself surrounded by driven individuals who are passionate about leveraging technology for impactful outcomes while enjoying benefits such as unlimited PTO and daily meals, all aimed at fostering a positive workplace atmosphere.

Join Rise to see the full answer
Common Interview Questions for Senior Software Engineer - ML/Data Infrastructure
Can you describe your experience with data pipelines and how you've built them in the past?

When answering this question, focus on specific examples where you've designed and implemented data pipelines, detailing the technologies and methodologies you used. Highlight any challenges you faced, how you overcame them, and the impact your work had on the project. Relate your experience to UnitX's mission and the type of data you’d expect to work with.

Join Rise to see the full answer
How do you ensure data quality and system observability in ML processes?

Address this question by describing the processes and frameworks you've implemented to monitor data quality and system performance in past projects. Discuss tools you have used for logging, monitoring, and alerting, and detail how you incorporate feedback loops in your data pipelines to maintain high quality as you iterate on models, emphasizing the continuous improvement aspect.

Join Rise to see the full answer
What are some of the best practices you've implemented for Machine Learning Operations (ML Ops)?

Share insights from your past experiences that showcase your understanding of ML Ops best practices. Discuss strategies you've used for versioning models, tracking experiments, promoting code quality, and CI/CD for ML applications. Be specific about the tools and platforms you've worked with, as this demonstrates your hands-on experience and thought leadership in this area.

Join Rise to see the full answer
How do you prioritize tasks when working on multiple projects simultaneously?

Use this opportunity to discuss your time management and organizational skills. Describe methods you employ to prioritize tasks based on project deadlines, dependencies, and impact, such as Agile methodologies or workflow management tools. Mention any experiences where your prioritization led to successful project outcomes, emphasizing how you adapt to fluid environments.

Join Rise to see the full answer
Can you give an example of how you've collaborated with cross-functional teams in a project?

Reflect on times when you've worked alongside different teams, such as engineering, product, or design. Describe how you facilitated communication, shared objectives, and resolved conflicts to achieve a common goal. Highlight the importance of effective collaboration and how it led to successful project deliveries, which is crucial at a team-focused company like UnitX.

Join Rise to see the full answer
What challenges have you faced in integrating ML solutions with real-time systems, and how did you handle them?

Talk about specific challenges such as latency issues, data synchronization, or hardware constraints. Share your analytical approach to solving these challenges, including any technologies or strategies that you employed. Relating these experiences to the specific needs of UnitX’s robotics products can demonstrate both your expertise and understanding of the job's challenges.

Join Rise to see the full answer
Describe your experience with cloud computing and how you've leveraged it in your previous roles.

In your response, showcase your experience with cloud providers (like AWS, Azure, or Google Cloud) and highlight specific services (like S3, Lambda, or Kubernetes) you utilized for deploying and managing ML applications. Discuss any architectural decisions you've made and how they've improved scalability, flexibility, or cost-efficiency for projects.

Join Rise to see the full answer
How do you stay updated with the latest developments in machine learning and data engineering?

Explain your approach to continuous learning, being sure to mention specific resources you turn to, such as journals, online courses, or communities. Discuss any personal projects or contributions to open-source that reflect your commitment to growing your skills. Staying up-to-date is pivotal in a rapidly evolving tech landscape, and it showcases your dedication to your craft.

Join Rise to see the full answer
What role do you believe mentorship plays in engineering, and how have you approached it?

This question allows you to express your views on mentorship, emphasizing its importance in fostering a collaborative environment. Share examples of how you've mentored junior staff or engaged in knowledge-sharing initiatives, and explain how this aligns with the culture of continuous improvement and high performance at UnitX.

Join Rise to see the full answer
Why do you want to work at UnitX as a Senior Software Engineer - ML/Data Infrastructure?

When answering this question, tailor your response to reflect an understanding of UnitX's mission and values. Discuss why the specific projects and technology at UnitX excite you and how they align with your own career goals and aspirations. Express enthusiasm for the opportunity to contribute to cutting-edge technology that reshapes manufacturing.

Join Rise to see the full answer
Similar Jobs
Photo of the Rise User
Posted 3 days ago

UnitX is seeking a proactive Product Manager for their AI and Imaging Software team to lead innovative product development efforts.

Photo of the Rise User
Posted 2 days ago

Join UnitX as a Product Manager specialized in 3D Robotics, driving innovative automation solutions in a rapidly evolving environment.

Photo of the Rise User
Posted 2 days ago

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 Remote Austin, Texas
Posted 11 days ago

Esper is looking for a Backend Engineer to elevate data-driven policymaking in government through innovative software solutions.

Posted 13 days ago

As a key player in our startup, you'll develop a cutting-edge simulation platform that empowers AI agents to perform software engineering tasks.

Weekday AI Remote No location specified
Posted 9 days ago

An exciting opportunity for an experienced SIP Developer to advance VoIP technologies in a flexible remote role.

Photo of the Rise User
Posted 12 days ago

As a Senior Engineer at Cardinal Health, you will lead SAP solutions for the Pharma business while working remotely.

Photo of the Rise User

Join Palo Alto Networks as a Sr Staff Engineer Software, where you will shape the future of cybersecurity through innovative cloud management solutions.

Photo of the Rise User
Posted 12 days ago

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.

Photo of the Rise User
Dare to be Different
Diversity of Opinions
Inclusive & Diverse
Customer-Centric
Transparent & Candid
Growth & Learning
Photo of the Rise User
Posted 10 months ago
Inclusive & Diverse
Rise from Within
Mission Driven
Diversity of Opinions
Work/Life Harmony
Photo of the Rise User
Inclusive & Diverse
Rise from Within
Mission Driven
Diversity of Opinions
Work/Life Harmony
Transparent & Candid
Growth & Learning
Fast-Paced
Collaboration over Competition
Take Risks
Friends Outside of Work
Passion for Exploration
Customer-Centric
Reward & Recognition
Feedback Forward
Rapid Growth
Medical Insurance
Paid Time-Off
Maternity Leave
Mental Health Resources
Equity
Paternity Leave
Fully Distributed
Flex-Friendly
Some Meals Provided
Snacks
Social Gatherings
Pet Friendly
Company Retreats
Dental Insurance
Life insurance
Health Savings Account (HSA)
MATCH
Calculating your matching score...
FUNDING
SENIORITY LEVEL REQUIREMENT
TEAM SIZE
EMPLOYMENT TYPE
Full-time, on-site
DATE POSTED
April 22, 2025

Subscribe to Rise newsletter

Risa star 🔮 Hi, I'm Risa! Your AI
Career Copilot
Want to see a list of jobs tailored to
you, just ask me below!