Earthformer

A Python package for Earth forecasting transformer

• Free software: Apache Software License 2.0
• Documentation: https://opengeos.github.io/earthformer

Introduction

This repository turns the amazon-science/earth-forecasting-transformer repository into a Python package, making it easier to install and use. To learn more about Earthformer, please refer to the NeurIPS 2022 paper: Earthformer: Exploring Space-Time Transformers for Earth System Forecasting. Credits to the authors and their repository for the original implementation.

Installation

The earthformer package is available on PyPI. To install it, run:

pip install earthformer

Earthformer has several optional dependencies. It is recommended to create a new conda environment and install the dependencies with conda:

conda create -n earthformer python=3.10
conda activate earthformer
conda install -c conda-forge mamba
mamba install -c conda-forge pytorch torchvision pytorch-lightning==1.7.7
mamba install -c conda-forge nvidia-apex omegaconf
mamba install -c conda-forge xarray netcdf4 opencv
pip install earthnet earthformer