Senior ML Engineer: Grid & Energy AI Platform

AZX; United States; Remote; USD 150,000 - 210,000; Full time

Benefits offered by this job

Health insurance

Equity

Bonus eligibility

Flexible paid time off

Fully remote culture with Seattle team

Job summary

AZX is building AI-powered solutions for climate and energy challenges, hiring a Senior ML Engineer to create reusable ML libraries and a building-energy simulation engine. You’ll work on forecasting, disaggregation, demand response, and asset health with Python and Rust, plus data engineering tasks and open-source tools.

As a fully remote, mission-driven company, AZX supports equity, healthcare, flexible PTO, and a Seattle cluster. Applicants authorized to work in the United States only.

Qualifications

• 5+ years shipping applied ML on real-world signals.
• Strong numerical and scientific computing skills.
• Python plus a systems language — the models and tooling are Python, the engine is Rust; depth in one, working ability in the other.
• 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 Python 3.12+ stack and time-series tooling.
• Comfort picking up Rust via PyO3/maturin and building evaluation harnesses and CI.
• Willingness to ramp quickly on energy-domain vocabulary.
• Bachelor’s Degree; Master’s is a plus

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, so client pods deploy proven components rather than reinventing them per engagement.
• Own the simulation engine — Rust crates and Python bindings — including its validation methodology and performance.
• Own the grid-data toolkits: protocol codecs, 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 with evidence attached.
• Own the feedback loop with client pods: turn field learnings into what to build next.

Skills

Forecasting

Disaggregation

Python

Rust

Evaluation harnesses

Data engineering

SQL/Postgres

PyO3/maturin

CI for scientific software

Education

Bachelor’s Degree

Master’s Degree preferred

Tools

PyO3/maturin

PostgreSQL

TimescaleDB

Job description

AZX is building AI-powered solutions for climate and energy challenges, hiring a Senior ML Engineer to create reusable ML libraries and a building-energy simulation engine. You’ll work on forecasting, disaggregation, demand response, and asset health with Python and Rust, plus data engineering tasks and open-source tools.

As a fully remote, mission-driven company, AZX supports equity, healthcare, flexible PTO, and a Seattle cluster. Applicants authorized to work in the United States only.