Senior Solutions Architect , ML

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.

The Opportunity:

Reporting into our Head of Customer Solutions, you will be responsible for partnering with our customer and internal stakeholders to ensure the technical feasibility and success of Union Cloud PoC’s and production deployments. We are looking for team members who are innately curious, get joy out of solving complex technical problems, and thrive in collaborative, self managed environments.


In this role, you will:

  • Own technical success of Union Cloud customers’ deployments and use cases by partnering with internal and customer stakeholders
  • Represent “voice of the user” and bring that user-first mindset across the company
  • Help Union Cloud users get up and running, and help them debug problems in their workflows
  • Empower our customers to attain efficiency, reliability, and repeatability for their ML applications, by educating them on good patterns
  • Partner with Product and Engineering teams to provide context and insights for feature design, solution and prioritization
  • Develop, maintain and document representative examples for Union Cloud customers
  • Learn, contribute to, and champion the Flyte open-source workflow orchestration system (see, as well as related Union Cloud
  • Educate technical users and decision-makers on Flyte and Union Cloud 
  • Maintain a cadence of communication with customers about their adoption trends, sentiment, and opportunities for deeper engagement
  • Support retention and adoption efforts of Union Cloud among customers by understanding their technical and business needs and helping them succeed

About you

  • Experienced with 5+ years of solving real-world ML problems to help users attain efficiency, reliability, and repeatability for their ML applications
  • Have solid Python programming skills. Java and / or Scala are pluses
  • You are familiar with public cloud technologies and services, DevOps/MLOps, networking and security principles, etc
  • Interested in a deeply technical customer facing role involving data and machine learning at scale
  • Significant experience with hands on machine learning and orchestration, building production-use ML systems in multi-tenant cloud products
  • In depth understanding of data processing, ML systems, MLOps and Machine Learning infrastructure components
  • Familiarity with Spark, Airflow, Tensorflow, Pytorch and other ML libraries
  • Possess excellent written and verbal communications skills, with a tone that conveys empathy
  • Have previous experience in a similar role supporting customers’ engineering teams adopting open source and commercial products for data and/or machine learning
  • It’s a strong plus if you have been a part of or contributed to the Flyte or Union communities through in-person meetups, conferences, workshops, events, electronic forums, or direct open-source development

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
  • Paid time off + 12 company holidays 
  • Paid sick days
  • 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
logo Union AI Technical Customer Support 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