Senior Python Engineer

About Us is a Seattle-based startup founded by a team of former Lyft engineers.  We contribute to Flyte,  an open source platform for orchestrating massively scalable machine learning pipelines running on Kubernetes. 

These pipelines power some of the most popular apps on your phone, as well as applications in diverse industries such as song sharing, vaccine research, early cancer detection, global map rendering, financial projecting, and piloting autonomous vehicles.

The Opportunity

Here at, we are an open source first company! We provide infrastructure for our customers to build delightful Machine Learning driven products and deliver them in production. As the lead Python engineer, you will be working on our Flyte OSS product. Our ideal candidate will get immense satisfaction out of knowing and upholding the intricacies and nuances of the Python language. This person will guide us in setting best practices around how our community and team members write performant Python code and tooling.

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

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

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

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

About You:

  • 5+ years as a backend engineer with 2+ years of experience in Python

  • Professional background in a Data science/engineering and ML

  • Experience building Python libraries and compilers

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

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

  • Open-source contributions in Python is a plus

  • Knowledge of Rust and PyO3 is a plus

Benefits & Belonging

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

Apply for this job
logo Union AI Engineering FullTime 🌎 Remote Apply Now
Your subscription could not be saved. Please try again.
Your subscription has been successful.


Subscribe and stay updated.

Your subscription could not be saved. Please try again.
Your subscription has been successful.

Join our newsletter