The Opportunity

Reporting into our Engineering Manager on Union Cloud, you will be responsible for building out new features and products across our paid product offering as we continue to onboard new customers to the platform.

In this role, you will:

• Contribute to the design of large scale, multi-tenant distributed control planes that manage complicated ML and Data Processes

• Design modern Golang based gRPC services

• Build core algorithms to help distributed ML and Data systems scale more easily

• Build core distributed scheduling systems on Kubernetes

• Build simple, elegant APIs that delight our customers - ML Engineers, Data Engineers, Data Scientists and Systems Administrators

• Manage terabytes of data in an efficient and secure manner

• Implement high performance Golang applications that can scale to millions of operations per second

• Work with the open source community and internal team members to deliver value across multiple organizations

About You:

• Bachelor’s Degree in Computer Science or related field, or equivalent experience

• Minimum 4+ years’ experience building scalable, performant distributed services in any strongly-typed language

• Proven discipline to check in code that is well-tested and well-documented

• Excellent verbal and written communication skills to ensure seamless integration with a global team and the open-source community

• Expertise in maintaining and debugging live software systems

• Ability to identify tasks which require automation and carry out the automation process

• Experience taking projects from inception to completion with minimal guidance, demonstrating curiosity and ownership

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

The Opportunity

Flytekit is the main entrypoint for users into the Flyte world. It's a Python SDK that serves as the gateway for the 'authoring' (writing tasks and workflows) part of Flyte. It's also the basis of the core tenet, the core mission of Union AI, and underscores why the Flyte project has found the success that it has.

We're looking for someone to join us and become the core maintainer of flytekit, with the goal of continuing to push on the mission of bringing as much ease-of-use and user-delight as possible while modernizing and elevating the existing code base. You will not only lead Python development at Union, but you will be a voice in the industry, leading other companies by example.

Prior work should have motivated you to develop a mastery of advanced Python topics - you should be able to seamlessly navigate things like asyncio and python threading, metaclasses, mros, module loading, and be able to use all this idiomatically, enumerating and explaining tradeoffs among the various ways of doing things, while integrating with and improving the flyte DSL.

In this role, you will:

• Lead the development of Python products at Union (e.g. flytekit, UnionML, pandera)

• Advocate for best practices across the Python language across our internal engineering members and external community contributors

• Go deep. Implement a DSL that mirrors the Python core language

• Get satisfaction out of understanding and solving our users’ ML and data challenges

• Since Union AI is focused heavily on machine learning and data applications, you should either bring with you or be willing to develop experience with ML libraries, learning and appreciating how AI practitioners use Python and the various ML toolchains that already exist

• Learn about new tools and libraries in the ML landscape, understanding their impact and use-cases, and implementing elegant integrations within Flyte

About You:

• 6+ years experience as a python backend engineer

• In-depth understanding of Python internals (CPython), lower-level programming, and infrastructure-related code development

• You have an interest in contributing to open source projects

• Bachelor’s Degree in Computer Science or related field, or equivalent experience

• Experience with ML Frameworks (Pytorch, Tensorflow, etc) and data processing frameworks (Spark, Flink, Beam etc) a plus

• Knowledge of Rust and PyO3 is a plus

• Open-source contributions in Python is a plus

• Knowledge of Rust and PyO3 is a plus

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

The Opportunity:

We are looking for a Technical Product Manager to join our team and take Union Cloud, our core managed product offering, to the next level of growth. This role involves working closely with early customers and our senior engineers to evolve the product into a fast, intuitive, and enterprise-ready AI development platform. We need someone who can develop a deep empathy for the pain our products solve for our users, can become an expert on Kubernetes-based developer tools, and has a clear vision of the product direction and how to get there.

In this role, you will:

• Own key areas of the Union Cloud platform and prioritize, plan, and innovate to drive business results

• Work as part of a hybrid team, in a cross-functional and collaborative role spanning product, design, development, documentation, QA, 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:

• 5+ years of Product Management experience

• Understanding of infrastructure concepts and tooling such as distributed systems, Kubernetes, and AWS/GCP offerings

• A track record of launching and iterating on products, bonus points if you have experience launching a product that failed

• Experience working with distributed compute, data, or machine learning workloads. Prior experience using or building workflow orchestration systems is considered a plus

• 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 experience

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

The Opportunity

Reporting into our Head of Customer Solutions, you will be responsible for partnering with our key customers and internal stakeholders to drive and implement technical strategy in select accounts. You will be on point to ensure the technical feasibility and success of Union Cloud customers through onboarding and beyond. This is a technical role, and we are looking for team members who are innately curious, get joy out of solving complex technical problems working alongside customers, and thrive in collaborative, self managed environments. Bonus points if you have been a part of or contributed to the Flyte/Union communities or contribute or other OSS.

In this role, you will:

• Develop reference architectures and best practices for optimal design, integration and operation of Union Cloud

• Collaborate with customers’ technical stakeholders to inform, debate and advise on technical architecture and solution topics

• Partner with our Engineering team to ensure a successful roll-out, onboarding and adoption of Union Cloud and Flyte for solving customer problems

• Represent “voice of the user” and bring that user-first mindset across the company

• Act as a customer advocate and main point of contact for select customers, with accountability for triage and resolution of major incidents alongside engineering

• Help Union Cloud users get up and running by debugging problems related to infrastructure, platform or code in customers’ integrated end-to-end solution

• 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

• Accelerate use case adoption and retention efforts of Union Cloud among customers by understanding their technical and business needs and helping them succeed

About you

• Have 5+ years of experience in deeply technical customer facing roles in sales engineering, solutions architecture, consulting or technical account management

• General understanding and practical knowledge of distributed systems at scale and familiarity with cloud platforms (AWS, GCP, Azure), Infrastructure as Code (Terraform, Cloudformation), Containers (Docker), and CI/CD systems

• Experience with technologies in the Kubernetes ecosystem is strongly preferred (Helm, Spinnaker, Prometheus, Envoy, Istio, etc)

• Experience with data processing, machine-learning systems, MLOps and machine learning infrastructure is strongly preferred

• Familiarity with popular machine learning frameworks (Pytorch, Tensorflow, etc) and data processing frameworks (Spark, Flink, Beam etc) is strongly preferred

• Possess excellent written and verbal communications skills, with a tone that conveys empathy

• Ability to engage at all levels of technical stakeholders to contribute and drive technical debates and design choices with our enterprise customers

• Previous experience in a similar role supporting customer engineering teams adopting commercial products for data and/or machine learning

• Are highly organized and thrive in a high-volume environment that will naturally have ambiguities and competing priorities

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