Union.ai · 1 month ago
Software Engineer (Distribution/Platform)
Union.ai is focused on solving challenges in AI infrastructure, particularly through their open-source project Flyte. They are looking for enthusiastic engineers to join their Distribution team, responsible for designing and implementing services for data and service orchestration in a large-scale AI/ML platform.
AI InfrastructureArtificial Intelligence (AI)Cloud InfrastructureIT InfrastructureMachine LearningProductivity ToolsSoftware
Responsibilities
Design and build backend services (APIs, Kubernetes controllers, etc) and client components to install, manage, and observe Union services in a Kubernetes native environment
Design, implement, and optimize distribution strategies to facilitate simple and intuitive management of a complex platform in customer controlled environments
Work across multiple cloud vendors including AWS, GCP, Azure, and OCI as well as neo-cloud providers
Develop and maintain services and tooling to make our systems more reliable, secure, and performant
Contribute to architectural decisions and participate in code and design reviews across various teams, ensuring the highest standards of quality and performance
Work closely with broader teams including Backend, Frontend, and Support to improve the experience for our customers
Qualification
Required
5+ years of experience in deeply technical roles in engineering functions
A deep passion for all things Kubernetes and the broader container orchestration ecosystem
Ability to navigate and pick up new technologies quickly
Ability to think about the big picture and can put yourself in the shoes of the developer and customer
Hands-on experience with backend programming languages (Go, Rust, Python)
Ability to own complex projects from planning to completion
Preferred
General understanding of building modern web applications using Next.js, React, and Typescript
Benefits
Excellent medical - We pay 100% of your premiums and 90% for your dependents
Generous dental and vision plans- We pay 90% of the premiums for you and your dependents
Meaningful equity in the form of options – all employees are owners here
Unlimited time off + 12 company holidays
401K match - Union.ai matches 100% of contributions up to the first 3%, and 50% up to 5%
16 weeks paid parental leave for primary and secondary caregivers
Flexible work schedule (some restrictions apply)
For in office employees: Lunch provided onsite and well stocked kitchen with snacks and drinks.
Company
Union.ai
Union.ai is a Kubernetes-native workflow orchestration platform for data and machine learning at scale.
H1B Sponsorship
Union.ai has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (5)
2022 (2)
Funding
Current Stage
Early StageTotal Funding
$29.1MKey Investors
New Enterprise Associates
2023-05-17Series A· $19.1M
2022-04-12Seed· $10M
Recent News
2025-12-09
Company data provided by crunchbase