Staff Software Engineer

About the Company

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. We steward Flyte, the open-source project that has emerged as the modern standard for data and AI orchestration.

Responsibilities

• Forge the future of Machine Learning (ML) tooling and enhance usability
• Steer technical direction, making strategic decisions on architecture, technologies, and processes
• Code, design, and architect high-quality, scalable, and maintainable solutions
• Craft robust APIs and deliver seamless user experiences across the platform
• Champion user-centric features by leveraging deep understanding of developer tools and frameworks
• Act as a technical liaison across teams and participate in code and design reviews
• Navigate and solve complex problems independently to drive projects to completion

Requirements

• Bachelor's degree in Engineering, Computer Science, or a related field
• 6 years of progressively responsible experience in IT or a related occupation
• Full-stack development experience with React, TypeScript, and JavaScript
• Backend experience with Python, Go, and Java
• Experience with Machine Learning frameworks including TensorFlow, PyTorch, and Spark
• Experience with data processing frameworks such as Spark, Flink, and Beam
• Proficiency with Rust and PyO3
• Experience with Python Internals and Flyte or open source contributions

Benefits

• 100% medical premium coverage for employees and 90% for dependents
• 90% dental and vision premium coverage for employees and dependents
• Meaningful equity in the form of options
• Unlimited time off plus 12 company holidays
• 401K match up to 5%
• 12 weeks paid parental leave
• Flexible work schedule
• Onsite lunch and well-stocked kitchen for in-office employees