Union.ai · 17 hours ago
AI Solutions Lead
United States
Full-time
Remote
Senior Level
$120K/yr - $170K/yr
5+ years expUnion.ai is focused on solving challenges in AI infrastructure and is seeking an AI Solutions Lead to act as the primary liaison for key customers. The role involves guiding customers through the adoption and optimization of the Union Cloud platform and Flyte, ensuring their success by leveraging expertise in AI workflows and data pipelines.
AI InfrastructureArtificial Intelligence (AI)Cloud InfrastructureIT InfrastructureMachine LearningProductivity ToolsSoftware
Responsibilities
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 expertise 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
Qualification
Required
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
Preferred
You've contributed to or participated in relevant AI or MLOps open-source communities, forums, or customer advocacy programs
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
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.
11-50 employees
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
Leadership Team
Ketan Umare
Co-Founder, CEO
Martin Stein
Investor - Union.ai - Generative AI and Machine Learning Workflows
Recent News
2025-12-09
Company data provided by crunchbase