Note: The job is a remote job and is open to candidates in USA. Vibrant Planet develops cloud-based, AI-driven technology and wildfire science solutions that help communities, land managers, utilities, and insurers reduce destructive wildfire risk. The ML Engineer will build, adapt, and operationalize geospatial foundation model-based deep learning systems for estimating forest structure metrics, integrate them into automated production pipelines, and maintain related data infrastructure. The role also contributes to scientific manuscripts and collaborates across science, data engineering, and product teams.
Responsibilities
- 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
- 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
- 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
- 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
Skills
- •• M.S. in Computer Science, Machine Learning, Remote Sensing, Data Science, Ecology, or a related quantitative field (or equivalent work experience)
- •• 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)
- •• 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
- Candidates must already be authorized to work in the U.S. without the need for sponsorship
- Location: •Remote, US | [Pacific / Mountain time zone preferred]•
- •• 3+ years of experience developing, training, and deploying deep learning models (PyTorch preferred)
- •• Experience with containerization (Docker) and familiarity with cloud platforms (AWS preferred)
- •• 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
Benefits
- 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)
- Remote-first work environment
Company Overview