ML Engineer

ABOUT THE ROLE

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.

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