Job Description

Join to apply for the Technical Account Manager role at Union.ai

Continue with Google Continue with Google

1 week ago Be among the first 25 applicants

Join to apply for the Technical Account Manager role at Union.ai

Get AI-powered advice on this job and more exclusive features.

Continue with Google Continue with Google

Continue with Google Continue with Google

Continue with Google Continue with Google

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.

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

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.

In this role, you will:

• 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 customers 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.
• Faciliate 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.

About You:

• 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 devtools or platforms.
• You’ve worked closely with enterprise engineering and product teams—whether in customer success, solutions engineering, DevRel, 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 genuine desire to learn, given its prevalence in data science.
• You’ve developed and executed account plans focused on growth through adoption, usage expansion, and product-led revenue—especially in a land-and-expand motion.
• You write clearly, speak thoughtfully, and can tailor your communication to both hands-on users and executive sponsors.
• You are curious, proactive, and self-directed—you seek ownership and are not afraid to experiment.
• You build trusted relationships quickly and are motivated by helping others succeed; you view the success of your customers as your own.
• Bonus: You’ve contributed to or participated in relevant AI or MLOps open-source communities, forums, or customer advocacy programs.

Nice to Have

• Direct experience with Flyte (open-source or Union).
• Experience working with large enterprise customers.
• Prior experience at a startup or fast-growing technology company.
• Experience in pre-sales or post-sales technical consulting.

Benefits & Belonging

At Union.ai 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 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.

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. Union.ai is an equal-opportunity employer and we encourage you to apply, even if your experience doesn’t align exactly with our job description.

Seniority level

• Seniority level

  Mid-Senior level

Employment type

• Employment type

  Full-time

Job function

• Job function

  Sales and Business Development

• Industries

  Software Development

Referrals increase your chances of interviewing at Union.ai by 2x

Get notified about new Technical Account Manager jobs in San Francisco, CA .

Regional Account Manager Foodservice, West

Regional Account Manager, LinkedIn Sales Solutions

San Francisco, CA $151,000 - $230,000 2 days ago

Client Account Manager, Mid-Market (Goods)

San Francisco, CA $163,000 - $221,000 2 weeks ago

Client Account Manager, Mid-Market (Services)

San Francisco, CA
$105,000.00
-
$147,000.00
18 hours ago

San Francisco, CA
$100,000.00
-
$120,000.00
6 days ago

Area Sales Manager - San Francisco - 21st Amendment Brewery Cafe, LLC

Strategic Account Manager, RL Eyewear Channel

Walmart eCommerce (Marketplace)- Strategic Account Manager

San Bruno, CA
$104,000.00
-
$202,000.00
2 weeks ago

Oakland, CA
$110,000.00
-
$140,000.00
1 day ago

San Francisco, CA $125,000 - $150,000 3 weeks ago

Dealership Account Manager - San Francisco, CA

Account Manager, Strategic Partnerships | California -Bay Area

San Francisco, CA $75,000 - $85,000 2 weeks ago

Account Manager - Fintech Named Accounts (San Francisco or New York)

Associate Account Manager (San Francisco, CA)

San Francisco, CA $77,500 - $108,500 6 days ago

We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.

#J-18808-Ljbffr

Job Tags

Full time, Work at office, Flexible hours,

Similar Jobs