The Role

Union AI is 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 at Union you are passionate about Python and will set the standard for its use and development within Union.


  • You will lead the development of Python products at Union (e.g. flytekit, unionml, pandera)
  • You enjoy writing performant Python code and tooling
  • You love understanding and solving our users’ ML and data challenges
  • You love learning about new tools and libraries in the ML landscape, understanding their impact and use-cases, and implementing elegant integrations within Flyte


  • 2+ years of experience in Python
  • Data science/engineering and ML
  • 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

Additional Skills

  • Experience working in an active open source community is a plus
  • Working knowledge of Kubernetes would helpful but not required


  • Fast-growing company with evolving challenges and opportunities to grow into almost any role you can fulfill
  • Meaningful equity -- we want all employees to be owners
  • Family-first culture with excellent medical, dental, and vision benefits
  • 401k with company match
  • Flexible work schedule, in the Seattle / Bellevue area or remote
  • Full Immigration Support including H1b transfers and sponsorship
  • Work with a pretty good crew of friendly nerds

And Last

We value diversity across many dimensions: age, gender, race, sexual orientation, faith, country of origin, economic background.  We want everyone to feel at home working here, and we know we are stronger when the team is composed of people who bring broad life experiences to the table. We especially value gumption, so overcoming challenges of any kind in your life will resonate with us.  If in doubt, reach out.

Apply for this Job

* Required
resume chosen  
(File types: pdf, doc, docx, txt, rtf)
cover_letter chosen  
(File types: pdf, doc, docx, txt, rtf)