Base Career helps you apply smarter for this job.
Key skills for this role
We're looking for an ML Engineer to own the technical backbone of how AZX serves and evaluates models at scale. This is a high-leverage IC role spanning our inference platform — GPU scheduling, autoscaling, and serving infrastructure for vLLM/SGLang across cloud and customer-managed clusters — and the evaluation systems that tell us whether model, prompt, and agent changes actually make things better.
You'll create technical direction for how AZX serves models reliably. This role suits someone who wants architectural ownership over hard ML infrastructure problems, paired with the judgment to build the guardrails that let the rest of the team move fast safely.
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 (Puget Sound Energy).
We’re a public benefit corporation, founded in 2024, and have been profitable from the beginning (bootstrapped with consulting).
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.
We're looking for an ML Engineer to own the technical backbone of how AZX serves and evaluates models at scale. This is a high-leverage IC role spanning our inference platform — GPU scheduling, autoscaling, and serving infrastructure for vLLM/SGLang across cloud and customer-managed clusters — and the evaluation systems that tell us whether model, prompt, and agent changes actually make things better.
You'll create technical direction for how AZX serves models reliably. This role suits someone who wants architectural ownership over hard ML infrastructure problems, paired with the judgment to build the guardrails that let the rest of the team move fast safely.
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
Seattle, USA
, USA
, USA
, USA
, USA
, USA
, USA
, USA
Remote, but only USA/Canada
Must be willing to travel to the Seattle area for the final interview and travel 2x/year for company summits
Additionally, depending on location, candidates can expect to spend 10-20% of their time working onsite with clients.
Applicants must be currently authorized to work in the United States on a full-time basis.
We are currently unable to sponsor or take over sponsorship of employment visas.
If this job sounds like a great fit, we’d love to hear from you. If you feel aligned with the company but don’t check ALL of these boxes, we’d still love to hear from you!
Public benefit corporation delivering custom AI transformation and engineering services to critical-industry organizations.
Visit company websiteJobs and hiring trendsUSD 140000-220000 yearly / year
Full-time
Senior · 3+ years experience
Remote
Apply faster on company sites with our extension.