Technical Product Manager, ML Engineer

About Us

At Union, we are solving one of the hardest challenges in AI infrastructure today: enabling high-velocity iteration while maintaining seamless production-readiness for AI workloads at scale. Flyte, the open-source project we steward, is the emerging standard for modern data and AI orchestration, with numerous leading technology organizations - like LinkedIn, Spotify, and Gojek - running millions of mission-critical workloads on the platform. We have a deep bench of infrastructure veterans from companies in the Big Three and beyond and a technical founding team who originally created Flyte while at Lyft.

We are looking for a Technical Product Manager with prior experience as an ML Engineer to join our team and take Union, our core managed product offering, to the next level of growth. This role involves working closely with early customers (ML and Data Engineers) and our senior engineers to evolve the product into a fast, intuitive, and enterprise-ready AI development platform. We need someone who has deep empathy for the pain our products solve for our users.

In this role, you will:

  • Own key areas of the Union platform and prioritize, plan, and innovate to drive business results forward
  • Advocate for Union via blogs, community syncs, webinars, conferences, and hackathons
  • Work as part of a hybrid team, in a cross-functional and collaborative role spanning product, design, development, documentation, marketing, sales, and support
  • Collaborate across the Union leadership team as well as with Product Management and Engineering
  • Lead frequent product discovery calls with customers to understand their needs, listen to their feedback, and distill patterns which will influence the product roadmap and feature design
  • Coordinate project execution across feature teams, adapting to new information and sometimes changing priorities

About you: 

  • 3+ years of Product Management experience
  • 2+ years experience as an ML Engineer
  • Experience working with distributed compute, data, or machine learning workloads 
  • You have an understanding of infrastructure concepts and tooling such as distributed systems, Kubernetes, and AWS/GCP offerings
  • A track record of launching and iterating on products (at least two cycles), bonus points if you have experience launching a product that failed 
  • Prior experience as a product manager for a B2B SaaS product - identifying customer needs, prioritizing among potential options, and delivering compelling products that delight users
  • Ability to thrive in the dynamic environment that comes with being part of a high-growth company. That means flexibility and willingness to jump in and do whatever it takes to be successful
  • Crisp and clear communicator who is able to align individuals at various technical levels
  • Bachelor's degree in Computer Science, Engineering or equivalent years of professional experience

Benefits & Belonging

At we know that employees who feel their best can build amazing things and we are proud to offer best in class benefits that will continually evolve and grow as the needs of our employees do. Benefits may vary based on country

  • Excellent medical - We pay 100% of your premiums and 90% for your dependents
  • Generous dental and vision plans- We pay 80% of the premiums for you and your dependents
  • Meaningful equity in the form of options – all employees are owners here
  • Unlimited PTO + 12 company holidays 
  • 401K match - matches 100% of contributions up to the first 3%, and 50% up to 5%
  • 12 weeks paid parental leave for primary and secondary caregivers
  • Flexible work schedule at a remote friendly company

We believe that our differences are what bring us together to achieve truly special outcomes. We strive to be inclusive and focus on building teams that embody that quality too. is an equal-opportunity employer and we encourage you to apply, even if your experience doesn’t align exactly with our job description.

Apply for this job

Other AI Jobs like this

logo Union AI Technical Product Manager FullTime 🌎 Remote Apply Now
Your subscription could not be saved. Please try again.
Your subscription has been successful.


Subscribe and stay updated.

Your subscription could not be saved. Please try again.
Your subscription has been successful.

Join our newsletter