xesn Documentation

xesn Documentation#

xesn is a python package for implementing Echo State Networks (ESNs), a particular form of Recurrent Neural Network originally introduced by Jaeger [2001]. The main purpose of the package is to enable ESNs for relatively large scale weather and climate applications, for example as by Smith et al. [2023] and Arcomano et al. [2020]. The package is designed to strike the balance between simplicity and flexibility, with a focus on implementing features that were shown to matter most by Platt et al. [2022].

xesn uses xarray to handle multi-dimensional data, relying on dask for parallelization and to handle datasets/networks that are too large for a single compute node. At its core, xesn uses numpy and cupy for efficient CPU and GPU deployment.