Technical Account Manager

Technical Account Manager

Apply for the Technical Account Manager role at Union.ai
. The position is currently open.

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 te, the open‑source project we steward, is the emerging standard for modern data and AI orchestration, and is used by leading technology organizations such as Linked In, Spotify, and Gojek. We have a deep bench of infrastructure veterans from companies in the Big Three and beyond, and a technical founding team that originally created Flyte while at Lyft.

The goal of the Technical Account Manager role is to be the primary liaison for our key customers, ensuring their successful adoption, implementation, and ongoing optimization of the Union Cloud platform and Flyte. Leveraging your deep expertise in AI workflows, data pipelines, and cloud environments, you will guide customers through complex integrations, provide training, troubleshoot challenges, drive product utilization, and translate their evolving business needs into actionable insights for our product and engineering teams.

You will ensure stakeholders get maximum value from the platform. Your ability to build strong cross‑functional relationships, anticipate requirements, and deliver proactive solutions will be pivotal in driving customer retention and expanding our platform’s value within their AI and data ecosystems.

You’ll act as a strategic partner to users, helping them navigate evolving AI workflows while surfacing key insights that inform product development and roadmap priorities.

You’ll collaborate directly with AI, ML and platform engineering teams, as well as technical decision‑makers, to ensure seamless implementation, accelerate value realization, and build strong partnerships that support retention, expansion, and customer advocacy.

• Own and drive strategic account plans focused on adoption milestones, usage expansion, and revenue growth in a land‑and‑expand model.
• Partner with internal technical teams to help customers design and optimize workflows for production‑grade AI systems.
• Own end‑to‑end customer outcomes and value from Union. Guide customers through onboarding and integration, troubleshoot challenges, drive product utilization, and provide timely, effective support to resolve complex issues.
• Translate real‑world challenges from customer environments into actionable feedback for our product and engineering teams.
• Track customer health, identify signs of friction or opportunity, and develop engagement strategies across both technical and executive stakeholders.
• Drive renewals and commercial growth by aligning our product capabilities with evolving customer needs and identifying upsell or cross‑sell pathways.
• Facilitate technical enablement content creation—reference architectures, debugging guides, cookbooks, and tutorials—to scale your impact.
• Lead customer touchpoints such as roadmap sessions, architecture reviews, QBRs, and executive briefings to ensure we remain aligned with customer goals and value realization.
• Serve as the internal voice of the customer in planning, prioritization, and go‑to‑market alignment.
• You will be expected to be in‑office (Seattle/Bellevue).

• You’re excited to work at the intersection of AI infrastructure and customer outcomes—and are driven to own both the technical and commercial success of your accounts.
• 5+ years of experience in a technical role such as Solutions Engineer, Technical Account Manager, Solutions Architect, or ML/Data Engineer within a B2B dev‑tools or platforms.
• You’ve worked closely with enterprise engineering and product teams—whether in customer success, solutions engineering, Dev Rel, or internal ML platform roles—and can credibly lead both architecture discussions and business reviews.
• You have at least design and architecture‑level understanding of systems involving data processing, model training, serving, or monitoring.
• Experience in MLOps/LLMOps, machine learning pipelines, and data orchestration tools.
• Proficiency in Python, or…