AZX · 4 months ago
Founding ML Engineer (Senior)
Seattle
Full-time
Onsite
Senior Level
$140K/yr - $220K/yr
5+ years expAZX is a public benefit corporation focused on accelerating positive impact in critical industries through AI transformation. The Founding ML Engineer will play a crucial role in client delivery and platform development, working on AI projects and collaborating with domain experts to create impactful ML solutions.
Computer Software
No H1B
Responsibilities
You will work on AI projects in client engagements and, over time, internal platform capabilities
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
Qualification
Required
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 production ML solutions
Experience with cloud ML platforms (AWS SageMaker, GCP Vertex AI, or Azure ML)
Champion for quality (model validation, reproducibility, monitoring, bias/variance checks)
High emotional intelligence and a learning mindset
Strong collaboration skills
Enjoy others' success and a fun, positive environment
Comfortable making decisions in the face of ambiguity and course correcting as needed
Preferred
Experience in both startup and enterprise environments
Past work in energy, real estate, utilities, climate, or related fields
Advanced ML/AI frameworks and techniques (e.g., PyTorch Lightning, JAX, HuggingFace, ONNX optimizations)
Lower-level or performance-focused languages for ML acceleration (e.g., C++, Rust, CUDA)
Large-scale data and distributed training paradigms (e.g., Spark, Ray, Horovod, Dask)
Advanced data infrastructure (e.g., vector/graph databases, feature stores, data lakes)
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
Company
AZX
Tech transformation, societal shifts and environmental disruption put enormous pressure on critical industries like energy, infrastructure, real estate and others to adapt, pursue new growth opportunities and optimize operations.
11-50 employees
Funding
Current Stage
Early StageCompany data provided by crunchbase