Senior ML Engineer (Energy & Utilities)

The role

You will build AI models and underlying tools for utility clients. This includes building a building-energy simulation engine, grid-data toolkits, and reusable ML libraries.

What you will do

• Own the reusable utility ML libraries including forecasting, disaggregation, demand response, detection, and asset health
• Build capabilities as tested libraries with validation harnesses
• Own the simulation engine including Rust crates and Python bindings
• Own the grid-data toolkits including protocol codecs, synthetic scenario generators, and verification utilities
• Build the planning and dispatch support behind demand-response programs
• Own the publish path for versioned crates and Python packages
• Own the feedback loop with client pods to improve capabilities based on field data

Required qualifications

• 5+ years of shipping applied ML on real-world signals
• Strong numerical and scientific computing skills
• Python plus a systems language
• Library craft
• Willingness to do your own data engineering
• Practical fluency in Python 3.12+ (numpy, pandas/polars, scikit-learn, statsmodels), time-series feature engineering, forecasting/clustering libraries (sktime/statsforecast-class), and SQL/Postgres or
• 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

Nice to have

• Master's is a plus

Benefits

• Competitive early-stage startup compensation
• Bonus eligibility
• Health insurance with meaningful coverage for dependents
• Flexible paid time off
• Equity

Before you apply

• 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
• Interview process includes a written take-home assignment followed by a live two-hour technical session
• Please only apply to a maximum of 2 roles at a time; applicants who apply to more than 2 roles within a 6 month period will automatically be disqualified

About the company

AZX specializes in physics-informed ML and enterprise AI solutions for climate and sustainability challenges. They work with category-leaders in real estate, energy, logistics and utilities.