Senior ML Engineer

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, and utilities for working with standardized building data formats — much of which we publish as open source, so some of the people using your work 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, and you'll join a specific project when your tools meet real-world data and need to be adjusted based on what actually happens in the field.

- 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, so client pods deploy proven components (like meter disaggregation or load forecasting with abstention and monitoring built in) 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, and use it to make city-scale building stock tractable through representative-archetype simulation.

- Own the grid-data toolkits: protocol codecs (IEEE C37.118-class), synthetic scenario generators, and verification utilities, including generating synthetic-but-believable meter data calibrated until domain experts can't tell.

- Build the planning and dispatch support behind demand-response programs, where acting on a wrong number carries real cost.

- Own the publish path — versioned crates and Python packages — shipped with the evidence (evals, benchmarks) 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.

- 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