Senior ML Engineer (Client Solutions)

Openkrill; United States; Remote; USD 140,000 - 210,000; Full time

Benefits offered by this job

Health insurance

Equity

Flexible PTO

Fully remote culture

Job summary

AZX is seeking an ML Engineer to deploy ML systems directly inside client environments. You will start with data across multiple systems and end with a model running on a schedule within the client environment.

You will wear many hats in a small team, spanning DevOps, infrastructure, and frontend/backend work, serving as the engineering face of AZX to clients. The role requires strong ML fundamentals and hands-on engineering excellence.

Qualifications

• 5+ years of shipping applied machine learning to production — forecasting, detection/classification on time series, survival/reliability modeling, or optimization.
• Strong data engineering skills to find, clean, join, and profile data at scale.
• Rigorous validation including chron. splits, walk-forward validation, and as-of correctness.

Responsibilities

• Own the full ML delivery lifecycle: data discovery, modeling, deployment in client environment, scheduling, monitoring, retraining policy.
• Build forecasting and detection models robust to real-world data quality issues.
• Backtest and defend precision/recall tradeoffs for operational decisions.
• Design systems to distinguish no prediction from wrong prediction.
• Ship usable product: FastAPI service, small React surface, scheduled job.
• Own the measurement story: baselines, KPIs, monitoring, post-deployment readouts.
• Maintain client-facing engineering presence and gather client data feedback.

Skills

5+ years of ML in production

Python

SQL

Docker

Time-series modeling

FastAPI

React/TypeScript

Education

Bachelor's Degree

Master's degree

Tools

PostgreSQL

TimescaleDB

Pandas

scikit-learn

Azure/AWS

Docker

FastAPI

React

Job description

About AZX

Our mission is to accelerate positive impact in critical industries through AI transformation. We specialize in physics-informed ML and enterprise AI solutions that directly address climate and sustainability challenges.

We’re growing quickly and already work with category-leaders in real estate (CBRE), energy (LevelTen Energy), logistics (Flexe) and utilities.

We bootstrapped profitably for our first year and are now backed by leading investors focused on AI, climate and energy.

We work on challenges in clean energy, decarbonization, climate risk, energy systems, and global economics. We’re building our company for long-term success and aim to create the ultimate place to work for those passionate about AI and making a positive impact.

About This Role

We are seeking an ML Engineer who builds ML systems directly inside client environments. Your job starts with the client's actual data spread across multiple systems — and ends with a model running on a schedule inside their environment. You will bring a strong area of expertise, but expect to wear many hats as part of a small team — some DevOps, some infrastructure, some front end and back end — because you are the engineering face of AZX to your client.

Responsibilities

• Own the full ML delivery lifecycle: data discovery and cleaning, modeling, evaluation, deployment into the client environment, scheduling, monitoring, and retraining policy.
• Build forecasting and detection models that hold up against real-world data quality issues (late feeds, revised rows, missing labels).
• Backtest and evaluate models honestly enough to stake real operational decisions on them, and defend your precision/recall tradeoffs to the people who bear the cost of false alarms.
• Design systems that distinguish "no prediction" from "wrong prediction," so a missing answer reads differently to the end user than an incorrect one.
• Ship enough product to make the model usable — a FastAPI service, a small React surface, a scheduled job — whatever "usable capability" means for that client.
• Own the measurement story: agree on baselines and KPIs before deployment, instrument for monitoring, and deliver a post-deployment readout with attribution limits clearly stated.
• Maintain client-facing engineering presence and a feedback loop into the platform team — running discovery, working sessions with client IT/data teams, demos, and surfacing the data shapes and failure modes only visible from inside client data.

Core Qualifications

• 5+ years of shipping applied machine learning to production — forecasting, detection/classification on time series, survival/reliability modeling, or optimization — with an evaluation you defended to someone whose job depended on it.
• Strong data engineering skills and willingness to use them: you find, clean, join, and profile data yourself at awkward scale, without a dedicated data team.
• Rigorous validation discipline — chronological splits, walk-forward validation, as-of correctness, and an instinct to be suspicious of a suspiciously good metric.
• Enough software engineering to ship real systems: Python, SQL, tests, Docker, a scheduler, an API or app surface, and monitoring — type-strict, tested, reviewable code, even in a pod of two.
• Client-facing capability and the assertion to use it — running discovery, leading demos, and pushing back early and plainly when an ask is wrong, with an alternative already in hand.
• Judgment about when ML is the wrong tool, and the willingness to say so to a client who wants AI regardless.
• Practical fluency with our core stack — Python 3.12+ (pandas/polars/DuckDB, scikit-learn, statsmodels, gradient boosting), SQL/Postgres (with TimescaleDB/PostGIS for grid work), and time-series feature engineering and validation.
• Comfort building the surfaces that make a model usable — FastAPI plus enough React/TypeScript to expose results — and deploying it with Docker and basic cloud tooling (Azure/AWS).
• Working fluency with LLMs for the agentic edges of client work (extraction, retrieval) — depth isn't required, but honesty about your actual experience is.
• Bachelor's Degree; Master's is a Plus
• Domain experience in Energy, Utilities, Infrastructure, and Commercial Real Estate is a plus

Why AZX!

• Be part of a fast-growing, profitable, mission-driven company with industry-leading clients tackling the massive opportunity of AI transformation in critical industries.
• 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

Additional Information

• 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, so if that format isn't a good fit, we'd ask that you not apply