Echo State Networks in JAX! Implements the ESN in JAX with a simple, sparse COO representation of the ESN reservoir. Usage Install the repo with pip install -e .[test] and run the test with pytest