A Python library to train neural network emulators of atmospheric transport models.

Installation

conda create -n neuraltransport python=3.10
conda activate neuraltransport
conda install -c conda-forge ffmpeg pkg-config libjpeg-turbo opencv cupy cuda-version=11.8
conda install -c conda-forge numpy pandas xesmf cdo python-cdo xarray dask zarr netCDF4 bottleneck matplotlib seaborn cartopy shapely xskillscore xrft pyarrow
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
conda install pyg -c pyg
pip3 install lightning cdsapi pypdf2 trimesh rtree ipykernel ipywidgets tensorboard einops timm ecmwf-api-client eccodes dm-tree cfgrib 
pip3 install git+https://github.com/jbusecke/xmovie.git
pip install pynvml wandb ruamel.yaml moviepy
pip3 install git+https://github.com/NVIDIA/modulus-makani.git
pip install -e .

Cite NeuralTransport

In case you use NeuralTransport in your research or work, it would be highly appreciated if you include a reference to our paper in any kind of publication.

@article{benson2024neuraltransport,
  title = {Atmospheric Transport Modeling of CO2 with Neural Networks},
  author = {Vitus Benson, Ana Bastos, Christian Reimers, Alexander J. Winkler,
 Fanny Yang and Markus Reichstein},
  eprint={2408.11032},
  archivePrefix={arXiv},
  primaryClass={cs.LG},
  url={https://arxiv.org/abs/2408.11032}, 
}

Contact

For questions or comments regarding the usage of this repository, please use the discussion section on Github. For bug reports and feature requests, please open an issue on GitHub. In special cases, you can reach out to Vitus (find his email on his website).