ML Engineer

ML Engineer POSITION DETAILS Department: Engineering Reports To: Engineering Manager, Infra Lead Location: Remote, US | [Pacific / Mountain time zone preferred] Employment: Full-time ABOUT VIBRANT PLANET We are a team of leaders in fire science, applied science, forestry, policy, and tech who work with land managers, community risk managers, utilities, and insurers to drive action that lowers the risk of destructive wildfire. Our cloud-based, AI-driven platform modernizes land management planning, community risk assessment, and monitoring through scenario building, decision support, and treatment outcome detection. Our fire science subsidiary, Pyrologix, produces leading wildfire science and models that quantify wildfire hazard and risk, and the benefits of mitigation action. This science powers the core Vibrant Planet platform and supports our work across utilities, insurance, and other sectors. Vibrant Planet is backed by climate and resilience leaders including Grantham Foundation, Earthshot, Elemental Excelerator, Ecosystem Integrity Fund, Cisco, and Halogen Ventures. For more information, visit vibrantplanet.net and pyrologix.com . ABOUT THE ROLE Vibrant Planet ( https://www.vibrantplanet.net/ ) harnesses data-driven science and cloud-based technology to help make communities and ecosystems more resilient in the face of global change. Our ML Engineering team sits at the intersection of machine learning, remote sensing, and forest ecology—building the models, pipelines, and data products that power our Land Tender decision-support platform. We are seeking a ML Engineer to build, adapt, and operationalize foundation model-based deep learning systems that estimate forest structure metrics from remotely sensed data. In this role you will fine-tune and adapt geospatial foundation models as a backbone to custom deep neural network heads, integrate trained models into Vibrant Planet’s automated production pipeline, and maintain the surrounding data infrastructure. You will also contribute to scientific knowledge dissemination through manuscripts and serve as a key cross-team link between SciDev and Data Engineering. KEY RESPONSIBILITIES ML Model Development & Adaptation • Adapt and fine-tune custom or publicly available geospatial foundation models as backbone architectures for domain-specific deep neural network heads that estimate forest structure metrics (canopy height, biomass, basal area, etc.). • Prepare, curate, and manage training datasets from remote sensing sources (Sentinel-2, Sentinel-1, Landsat, lidar, NAIP) and field plot inventories. • Evaluate model performance using standard remote sensing accuracy metrics and field-based validation data. • Contribute to experiment design, hyperparameter optimization, and ablation studies in coordination with the Technical Lead ML Engineer. Pipeline & Data Engineering • Integrate trained ML models into Vibrant Planet’s automated geospatial data pipeline as containerized, orchestrated inference services. • Build and maintain STAC (SpatioTemporal Asset Catalog) infrastructure for data discovery, cataloging, and access control of ML model inputs and outputs. • Design and implement larger pipelines composed of many smaller DAGs (Airflow), ensuring idempotency, observability, and fault tolerance. • Maintain and improve data ingestion, preprocessing, and quality control workflows for satellite imagery and ancillary datasets. • Monitor pipeline health and model drift; implement alerting and automated retraining triggers as needed. • Develop model cards for summarization of modeling methods and performance. Knowledge Dissemination & Cross-Team Collaboration • Write and contribute to scientific manuscripts describing methods, validation results, and novel applications. • Serve as a cross-team link between SciDev, Data Engineering, and Product—translating requirements, communicating constraints, and aligning priorities. • Document pipelines, model architectures, and operational procedures in team knowledge bases. • Participate in code reviews, architectural discussions, and sprint planning. Team and Collaboration • Demonstrated ability to work collaboratively in interdisciplinary teams spanning science, engineering, and product. • Strong organizational skills to ensure high-quality data and clear documentation of workflows. • Ability to self-motivate, manage time, and work independently in a remote-first environment. • Excellent adaptive communication skills—ability to translate between scientific and engineering audiences. • Commitment to an inclusive and equitable work environment where diverse views and backgrounds are valued. • Comply with Vibrant Planet’s Information Security Policy and the full security responsibilities detailed in the Employee Handbook, including complete required security training, safeguard customer and company data, keep credentials secure, and report suspected security incidents or policy violations through established channels. • Follow secure development practices, adhere to established change management processes for production systems, protect the confidentiality and integrity of customer data, and promptly address security vulnerabilities in your area of responsibility. REQUIRED QUALIFICATIONS • M.S. in Computer Science, Machine Learning, Remote Sensing, Data Science, Ecology, or a related quantitative field (or equivalent work experience). • 3+ years of experience developing, training, and deploying deep learning models (PyTorch preferred). • Strong Python proficiency including data science stack (NumPy, pandas, xarray, scikit-learn). • 3+ years of experience with geospatial data processing (rasterio, GDAL, geopandas, shapely). • Experience building and maintaining data pipelines with workflow orchestration tools (Airflow, Prefect, Dagster, or equivalent). • Proficiency with Git, GitHub, and collaborative software development practices (code review, CI/CD). • Experience with containerization (Docker) and familiarity with cloud platforms (AWS preferred). • Familiarity with STAC specifications and geospatial data catalog infrastructure. • Strong written communication skills; ability to contribute to scientific manuscripts and technical documentation. • Basic knowledge of forest ecology, remote sensing principles, or natural resource science. PREFERRED QUALIFICATIONS • Ph.D. in a relevant field. • Experience with geospatial foundation models and self-supervised learning. • Experience with Kubernetes and distributed computing for large-scale inference. • Familiarity with ML experiment tracking (MLflow, W&B) and model registry practices. • Experience with database systems (PostgreSQL, PostGIS) and message queues. • Publications in remote sensing, ML, or ecology journals. COMPENSATION & BENEFITS Salary Range: $100,000 – $200,000 • Health, dental, and vision insurance • 401(k) plan • Unlimited PTO policy • Company equity • Cell phone stipend (per pay period) • Home office setup allowance (one-time) EQUAL OPPORTUNITY EMPLOYER Vibrant Planet is committed to diversity. We encourage applicants from all cultures, races, colors, religions, sexes, national or regional origins, ages, disability status, sexual orientation, gender identity, military, or other status protected by law to apply. We are most interested in finding the best candidate for the job, and that candidate may come from a less traditional background, but have capacity to grow into and thrive in the position after some mentoring. We do not require that you have experience with every job description task. We will consider any equivalent combination of knowledge, skills, education, and experience to meet minimum qualifications. We encourage each candidate to think broadly about their unique background and skill set and how it may relate to the role. While we welcome applicants from all backgrounds, we regret that we are unable to provide visa sponsorship (including H-1B) at this time. Candidates must already be authorized to work in the U.S. without the need for sponsorship.