[Hiring] Senior ML Engineer @AZX

Role Description

We are seeking a Senior ML Engineer that will build the AI models and underlying tools that power AZX's work with utility clients — reusable capabilities used across many client engagements. Underneath the models, you'll also build the infrastructure that makes them possible:

• A fast building-energy simulation engine
• Tools for reading real-time grid sensor data
• Utilities for working with standardized building data formats

Much of this work is published as open source, meaning some of the users are outside engineers you'll never meet. Rather than being assigned to one client account, you'll build the capabilities that every client-facing team draws on, joining specific projects when your tools meet real-world data and need adjustments based on field conditions.

Responsibilities

• Own the reusable utility ML libraries — forecasting, disaggregation, demand response, detection, and asset health — each shipped with its own evaluation harness and documentation.
• Build capabilities once as tested libraries with validation harnesses, allowing client pods to deploy proven components instead of reinventing them per engagement.
• Own the simulation engine — Rust crates and Python bindings — including its validation methodology against the reference oracle and its performance.
• Own the grid-data toolkits: protocol codecs (IEEE C37.118-class), synthetic scenario generators, and verification utilities.
• Build the planning and dispatch support behind demand-response programs.
• Own the publish path — versioned crates and Python packages — shipped with the evidence attached.
• Own the feedback loop with client pods: track what the capability got wrong in the field and turn that into what you build next.

Qualifications

• 5+ years of shipping applied ML on real-world signals — forecasting, disaggregation, detection/classification on interval or sensor data, survival/reliability modeling, or an adjacent-industry equivalent.
• Strong numerical and scientific computing skills: feature engineering from raw interval data, solver-level numerics when needed, and a healthy distrust of your own metrics.
• Python plus a systems language — depth in one, working ability in the other, and the appetite to close the gap.
• Library craft: you build things other engineers consume — versioned, tested, documented, with an API you'd want to call yourself.
• Willingness to do your own data engineering — finding, cleaning, joining, and profiling inputs yourself.
• Practical fluency in our core stack — Python 3.12+ (numpy, pandas/polars, scikit-learn, statsmodels), time-series feature engineering, forecasting/clustering libraries, and SQL/Postgres or TimescaleDB-class hypertables.
• Comfort picking up Rust (or a comparable systems language) via PyO3/maturin, and building evaluation harnesses and CI for scientific software.
• Willingness to ramp quickly on energy-domain vocabulary if you don't already have it.
• Bachelor's Degree; Master's is a plus.

Requirements

• Must be able to travel 2x/year for company summits.
• Applicants must be currently authorized to work in the United States on a full-time basis.
• We are unable to sponsor or take over sponsorship of employment visas at this time.
• Please note that our interview process includes a written take-home assignment followed by a live two-hour technical session with our engineering team.
• Please only apply to a maximum of 2 roles at a time; any applicants who apply to more than 2 roles within a 6 month period will automatically be disqualified.

Benefits

• Be part of a fast-growing, profitable, mission-driven company with industry-leading clients.
• Competitive early-stage startup compensation (based on capabilities, experience, and location).
• Bonus eligibility.
• Health insurance with meaningful coverage for dependents.
• Flexible paid time off.
• Equity.
• Fully remote culture with a cluster of teammates in Seattle.

Next Steps

If this job sounds like a great fit but you don’t check ALL of these qualification boxes, we’d still love to hear from you!