Echo State Networks powered by xarray
Project description
xesn
Echo State Networks powered by xarray and dask.
Description
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.
Installation
Installation from conda-forge
conda install -c conda-forge xesn
Installation from pip
pip install xesn
Installation from source
git clone https://github.com/timothyas/xesn.git
cd xesn
pip install -e .
Note that additional dependencies can be installed to run the unit test suite::
pip install -e .[test]
pytest xesn/test/*.py
Getting Started
To learn how to use xesn, check out the documentation here
Get in touch
Report bugs, suggest features, or view the source code on GitHub.
License and Copyright
xesn is licensed under the Apache-2.0 License.
Development occurs on GitHub at https://github.com/timothyas/xesn.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.