ML Engineer (Senior)

Core responsibilities:

• Turn ambiguous client problems into shipping code
• Drive projects from discovery to deployment
• 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

This role accelerates positive impact in critical industries through AI transformation, directly addressing climate and sustainability challenges by working on clean energy, decarbonization, climate risk, and energy systems.

Required qualifications:

• 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 (prompt engineering, RAG, fine-tuning, LangChain, model evaluation tooling)
• MLOps and infrastructure automation (CI/CD for ML, Docker, Kubernetes, Terraform, MLflow, Kubeflow)
• Experience with cloud ML platforms (AWS SageMaker, GCP Vertex AI, or Azure ML)

Compensation and benefits:

• 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
• Training and learning opportunities

AZX is a public benefit corporation whose mission is to accelerate positive impact in critical industries through AI transformation, specializing in physics-informed ML and enterprise AI solutions that directly address climate and sustainability challenges.