Back to jobs
New

Senior Machine Learning Engineer

United States
  • Deep expertise in building and operating production-grade ML and data platforms using Spark (PySpark + SQL), Databricks (Azure), and Delta Lake, with strong hands-on experience in MLOps practices including MLflow model lifecycle management, feature store architecture (offline + online), CI/CD for ML workflows, and scalable model deployment. Proven ability to design reliable, cost-efficient distributed data systems, optimize Spark workloads, and implement robust governance, observability, and access controls across ML data pipelines. Strong cloud engineering fundamentals in Azure, including orchestration, infrastructure reliability, and integration with services such as CosmosDB and downstream analytics systems.
  • The Data Engineering and Machine Learning teams at Zip exist to make data and ML production-ready, trusted, and scalable across the business. Our mission is to elevate the quality, reliability, and accessibility of data assets while enabling innovative AI-driven applications that create measurable customer and commercial impact.
  • We operate with an ownership mindset — engineers here don’t just build pipelines, they own platforms end-to-end. Great talent on this team thrives in ambiguity, designs with scale and reliability in mind, and proactively improves standards rather than maintaining the status quo. We work collaboratively across Data Science, Analytics, and Engineering, balancing speed with engineering discipline. We value pragmatic problem-solvers who think in systems, prioritize observability and maintainability, and are motivated by building infrastructure that empowers others to move faster and smarter.

Start your adventure with Zip 

We’re hiring a Senior Machine Learning Platform Engineer to build and operate the infrastructure that powers production-grade machine learning at Zip. In this role, you’ll own the ML lifecycle end-to-end — from feature pipelines and model registry standards to CI/CD and scalable model serving on Databricks (Azure). You’ll ensure our ML systems are reliable, observable, and built to scale as we expand AI-driven capabilities across the business.

Our goals include enhancing the discipline within our data engineering practices, strengthening our collaboration with the Data Analytics and Data Science teams, and elevating the quality of our data assets. These changes are designed to better position us to leverage the full potential of our data, allowing us to explore new and innovative applications, including the use of AI.

Interesting problems you’ll get to solve

Own the ML Lifecycle (MLOps)

  • Build and maintain feature pipelines (batch + streaming)
  • Manage offline and online feature store patterns (CosmosDB-backed online lookup)
  • Administer and enforce standards around MLflow model registry, versioning, and promotion workflows
  • Deploy and operate model serving endpoints
  • Implement CI/CD for ML pipelines and model deployment
  • Participate in on-call rotation for platform-owned systems

Build Production-Grade Spark Systems

  • Develop pipelines using PySpark and Spark SQL
  • Optimize joins, partitioning, and shuffle-heavy workloads
  • Improve reliability and cost-efficiency of Spark jobs
  • Ensure pipelines are modular, testable, and production-ready
  • Support streaming workloads using Delta Live Tables

Operate and Improve the Platform

  • Administer Databricks clusters, jobs, policies, and permissions
  • Improve observability, alerting, and operational standards
  • Contribute to Lakehouse Federation initiatives (Databricks ↔ Snowflake via Iceberg)
  • Integrate ML services into downstream architecture
  • Implement governance, access controls (RBAC), and data quality/observability standards across ML data pipelines

What you’ll bring to the team 

Experience

  • 8+ years of experience in Machine Learning with a strong focus on production-grade ML and distributed data systems
  • Demonstrated experience owning and operating ML systems end-to-end in production environments

Strong Spark Capability (Core Requirement)

  • Advanced experience with PySpark and Spark SQL
  • Strong understanding of Spark execution (joins, shuffles, partitioning)
  • Experience building and optimizing reliable, scalable data pipelines
  • Strong data engineering fundamentals including medallion architecture design, incremental/idempotent ETL patterns, and Delta Lake optimization (partitioning)

MLOps & ML Systems

  • Experience operating ML systems in production
  • Hands-on experience with MLflow (tracking + model registry)
  • Experience managing feature stores (offline + online)
  • Experience deploying and monitoring model serving endpoints
  • Experience implementing CI/CD for ML workflows

Cloud & Platform Experience

  • Experience working in Azure
  • Production experience with Databricks and Delta Lake
  • Experience integrating with CosmosDB or similar NoSQL key-value stores
  • Experience designing orchestrated, production-grade data workflows (Databricks Workflows, Airflow, or ADF) with dependency management, backfills, and failure recovery

Nice to Have

  • Delta Live Tables and streaming pipelines
  • Iceberg or Lakehouse Federation experience
  • Snowflake experience
  • Vector databases or LLM infrastructure
  • Infrastructure-as-code experience

What you’ll get in return

Zip is a place where you’ll get out what you put in. The newness of our sector means we need to move at pace and embrace change, and our promise to you when you join the team is that you’ll feel empowered and trusted to make big things happen quickly. 

We want you to feel welcome and as though you have the support to be yourself, and care for yourself at work. Because it’s important to us that you make the most of the opportunities you’ll get to grow your skills and your career, and be surrounded by smart, friendly people and leaders that have your back.

We think these are just some of the best things about being a Zipster. We will also offer you:

  • Flexible working culture
  • Incentive programs
  • Unlimited PTO
  • Generous paid parental leave
  • Leading family support policies
  • Company-sponsored 401k match
  • Learning and wellness subscription stipend
  • Beautiful Union Square office with a casual dress code
  • Industry-leading, employer-sponsored insurance for you and your dependents, with several 100% Zip-covered choices available

#LI-Remote

Zip is committed to a straightforward and transparent pay structure. The actual base salary will be determined by various individualized factors, including job-related knowledge, skills, experience, location, internal equity, as well as other objective business considerations.

The annual base Pay Range for this position is $150,000- 161,000K. This range reflects our US national compensation band (USN). Additional premium percentages may apply based on our tiered premium strategy.

Subject to those same considerations, the total compensation package for this position may also include other elements, including a bonus and/or commission awards, in addition to a full range of medical, financial, and/or other benefits. 

If hired, employees will be in an 'at-will position' and the Company reserves the right to modify base salary (as well as any other discretionary payment or compensation or benefit program) at any time, including for reasons related to individual performance, Company or individual department/team performance, and market factors.

Be a part of a team that reflects the diversity of our customers

We pride ourselves on being a workplace that provides equal opportunities to people of all ages, cultural backgrounds, sexual orientations, gender identities, abilities, veteran status, and everything else that makes you unique.

Equally, we’re committed to ensuring our recruitment processes are accessible and inclusive. Please let us know If there are any adjustments that need to be made to ensure you have a fair and equitable experience.

And finally…get to know us

Zip Co Limited (ASX: ZIP) is a digital financial services company, offering innovative, people-centered products.

Operating in two core markets - Australia and New Zealand (ANZ) and the US, Zip offers access to point-of-sale credit and digital payment services, connecting millions of customers with its global network of tens of thousands of merchants. We’re proud to be a values-led business and our values - Customer First, Own it, Stronger Together and Change the Game - guide us in everything we do.

I acknowledge by clicking "Submit Application", that the information provided is true and correct. I also understand that any willful dishonesty may render for refusal of this application or immediate termination of employment. By providing your information, you acknowledge that you have read our Zip Applicant and Candidate Privacy Notice and authorize Zip to process your data subject to those terms. Zip participates in the federal government’s E-Verify program.

Before you apply, give Zip a try   -> rebrand.ly/check-zip-out

Create a Job Alert

Interested in building your career at Zip Co Limited? Get future opportunities sent straight to your email.

Apply for this job

*

indicates a required field

Resume/CV*

Accepted file types: pdf, doc, docx, txt, rtf

Cover Letter

Accepted file types: pdf, doc, docx, txt, rtf

Select...
Select...
Select...

Read up on our candidate privacy statement here.

Voluntary Self-Identification

For government reporting purposes, we ask candidates to respond to the below self-identification survey. Completion of the form is entirely voluntary. Whatever your decision, it will not be considered in the hiring process or thereafter. Any information that you do provide will be recorded and maintained in a confidential file.

As set forth in Zip Co Limited’s Equal Employment Opportunity policy, we do not discriminate on the basis of any protected group status under any applicable law.

Select...
Select...
Race & Ethnicity Definitions

If you believe you belong to any of the categories of protected veterans listed below, please indicate by making the appropriate selection. As a government contractor subject to the Vietnam Era Veterans Readjustment Assistance Act (VEVRAA), we request this information in order to measure the effectiveness of the outreach and positive recruitment efforts we undertake pursuant to VEVRAA. Classification of protected categories is as follows:

A "disabled veteran" is one of the following: a veteran of the U.S. military, ground, naval or air service who is entitled to compensation (or who but for the receipt of military retired pay would be entitled to compensation) under laws administered by the Secretary of Veterans Affairs; or a person who was discharged or released from active duty because of a service-connected disability.

A "recently separated veteran" means any veteran during the three-year period beginning on the date of such veteran's discharge or release from active duty in the U.S. military, ground, naval, or air service.

An "active duty wartime or campaign badge veteran" means a veteran who served on active duty in the U.S. military, ground, naval or air service during a war, or in a campaign or expedition for which a campaign badge has been authorized under the laws administered by the Department of Defense.

An "Armed forces service medal veteran" means a veteran who, while serving on active duty in the U.S. military, ground, naval or air service, participated in a United States military operation for which an Armed Forces service medal was awarded pursuant to Executive Order 12985.

Select...

Voluntary Self-Identification of Disability

Form CC-305
Page 1 of 1
OMB Control Number 1250-0005
Expires 04/30/2026

Why are you being asked to complete this form?

We are a federal contractor or subcontractor. The law requires us to provide equal employment opportunity to qualified people with disabilities. We have a goal of having at least 7% of our workers as people with disabilities. The law says we must measure our progress towards this goal. To do this, we must ask applicants and employees if they have a disability or have ever had one. People can become disabled, so we need to ask this question at least every five years.

Completing this form is voluntary, and we hope that you will choose to do so. Your answer is confidential. No one who makes hiring decisions will see it. Your decision to complete the form and your answer will not harm you in any way. If you want to learn more about the law or this form, visit the U.S. Department of Labor’s Office of Federal Contract Compliance Programs (OFCCP) website at www.dol.gov/ofccp.

How do you know if you have a disability?

A disability is a condition that substantially limits one or more of your “major life activities.” If you have or have ever had such a condition, you are a person with a disability. Disabilities include, but are not limited to:

  • Alcohol or other substance use disorder (not currently using drugs illegally)
  • Autoimmune disorder, for example, lupus, fibromyalgia, rheumatoid arthritis, HIV/AIDS
  • Blind or low vision
  • Cancer (past or present)
  • Cardiovascular or heart disease
  • Celiac disease
  • Cerebral palsy
  • Deaf or serious difficulty hearing
  • Diabetes
  • Disfigurement, for example, disfigurement caused by burns, wounds, accidents, or congenital disorders
  • Epilepsy or other seizure disorder
  • Gastrointestinal disorders, for example, Crohn's Disease, irritable bowel syndrome
  • Intellectual or developmental disability
  • Mental health conditions, for example, depression, bipolar disorder, anxiety disorder, schizophrenia, PTSD
  • Missing limbs or partially missing limbs
  • Mobility impairment, benefiting from the use of a wheelchair, scooter, walker, leg brace(s) and/or other supports
  • Nervous system condition, for example, migraine headaches, Parkinson’s disease, multiple sclerosis (MS)
  • Neurodivergence, for example, attention-deficit/hyperactivity disorder (ADHD), autism spectrum disorder, dyslexia, dyspraxia, other learning disabilities
  • Partial or complete paralysis (any cause)
  • Pulmonary or respiratory conditions, for example, tuberculosis, asthma, emphysema
  • Short stature (dwarfism)
  • Traumatic brain injury
Select...

PUBLIC BURDEN STATEMENT: According to the Paperwork Reduction Act of 1995 no persons are required to respond to a collection of information unless such collection displays a valid OMB control number. This survey should take about 5 minutes to complete.