Bellevue, Washington, United States · Hybrid
$212k–$270k/yr
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-sour…
Skills: React, TypeScript, JavaScript, Python, Go
Sign up with Clera and we'll reach out the moment a role actually fits you — no more spraying applications into the void.
Lead the technical direction and architecture for AI infrastructure and ML tooling to enhance usability and scalability. Develop high-quality full-stack solutions and robust APIs while collaborating across teams to solve complex large-scale problems.
Requires a Bachelor's degree in Engineering or Computer Science with at least 6 years of experience in full-stack development and ML frameworks. Must have specific expertise in Python internals, Rust, PyO3, and data processing frameworks like Spark or Flink.
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, has emerged as the modern standard for data and AI orchestration, and is trusted by leading technology organizations including LinkedIn, Stripe, and Wayve to run millions of mission-critical workflows on the platform. These workflows comprise data preparation, model training, and scaled inference spanning thousands of GPUs, all major clouds, and on-premise infrastructure.
We have a technical founding team who created Flyte while at Lyft, a deep bench of infrastructure experts from top companies, and have raised from top investors like NEA and Nava Ventures.
Duties: Forge the future of Machine Learning (ML) tooling, infusing unique insights into feature development and enhancing usability. Solve complex, large-scale problems. Steer our technical direction, making strategic decisions on architecture, technologies, and processes. Code, design, and architect, establishing the benchmark for quality and innovation within the team. Utilize comprehensive expertise in both front-end and back-end development to engineer high-quality, scalable, and maintainable solutions. Craft robust APIs and deliver seamless user experience across our platform. Champion the creation of user-centric features by leveraging deep understanding of developer tools and frameworks. Address the needs and challenges of data scientists and engineers, advocating for user- centric design and functionality at every development phase. Act as a technical liaison across teams, enriching collaborative efforts with insights and expertise. Participate actively in code and design reviews, ensuring adherence to the highest standards of code quality and performance. Exhibit a proactive mindset with a focus on impactful work, addressing urgent needs within the team. Navigate and solve complex problems independently, driving projects to completion.
REQUIREMENTS: Bachelor's degree or equivalent in Engineering (any), Computer Science, or related; and 6 years of progressively responsible experience in the job offered or a related IT occupation. Must include 6 years of experience in each of the following: Using full-stack development with hands-on experience in React, TypeScript, JavaScript and a backend technologies including Python, Go and Java; Working with Machine Learning concepts and frameworks, including TensorFlow, PyTorch, and Spark; Bridging traditional software development with cutting-edge ML applications; Utilizing Python Internals; Utilizing Flyte or contributing to open source technologies; Utilizing data processing frameworks, including Spark, Flink, and Beam; and Working with Rust and PyO3.
OTHER:
Job Site: Bellevue, WA;
40 hours/week;
Salary: $212,202 - $270,000 per year.
Requisition Number: EXT9504598
If interested apply online at: https://www.union.ai/careers?ashby_jid=c131946e-640c-4885-ace7-f87912034489#positions
And reference the job title of the role and requisition number.
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%
12 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.
The enterprise Flyte platform for AI development and orchestration.
Unify data, models, and compute with the fastest path to a production-ready AI system. Orchestrate extremely scalable AI, ML, and agentic systems. Bringing you Flyte, the OSS standard for pipeline orchestration (Flyte.org) Twitter @union_ai Slack flyte-org.slack.com GitHub github.com/flyteorg/flyte YouTube youtube.com/@union-ai
Offices: Seattle, WA, US
The enterprise Flyte platform for AI development and orchestration.
Unify data, models, and compute with the fastest path to a production-ready AI system. Orchestrate extremely scalable AI, ML, and agentic systems. Bringing you Flyte, the OSS standard for pipeline orchestration (Flyte.org) Twitter @union_ai Slack flyte-org.slack.com GitHub github.com/flyteorg/flyte YouTube youtube.com/@union-ai
Offices: Seattle, WA, US
