ML Engineer (Senior)

## About the role ## 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 (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. ## About the Role We seek a Machine Learning (ML) Software Engineer to help us deliver for our clients and invest in our platform. You'll be one of the first full-time ML engineers and will have an enormous impact on all aspects of what we do. You'll join a company founded by experienced, successful veterans in climate and AI. The team brings exceptional expertise from Microsoft, AI2, ARM, acquired startups, and even a major utility, including PhDs, published ML researchers, and former industry executives. The role is initially 70% client delivery and 30% platform development, shifting to 50/50 over the first year. Some of our client projects are enterprise-style, and some are fast innovation cycles. Over time, we'll be investing more in our own platform to accelerate client value. ## What you will do You will work on AI projects in client engagements and, over time, internal platform capabilities. You will: - Turn ambiguous client problems into shipping code. - Drive projects from discovery to deployment - Collaborate with client and internal project teams - Design and write clean, scalable code at appropriate quality standards (sometimes "right", and sometimes "right now"). - Design, integrate, and productionize ML solutions including predictive models, GenAI systems, physics-informed ML, and digital twins. - Collaborate with domain experts in energy, real estate, and climate to translate business needs into ML solutions - Advocate for engineering best practices and positive dev culture ## Core Qualifications - Technical and foundational - 5+ years building and deploying ML systems in production environments - Expert-level Python and experience with PyTorch / TensorFlow - Deep expertise in at least one domain: NLP, Computer Vision, Time-Series, or Reinforcement Learning - Generative AI and LLM-related capabilities (e.g., prompt engineering, RAG, fine-tuning, LangChain, model evaluation tooling) - MLOps and infrastructure automation (e.g., CI/CD for ML, Docker, Kubernetes, Terraform, MLflow, Kubeflow) - Strong engineering fundamentals: system design, scalability, testing, and monitoring - Track record of translating ambiguous business problems into technical solutions