Resolve gpu memory issue with JAX esn #13
Description
When using very large reservoirs (>5000) in the ESN, jax allocates too much memory and the code fails. The issue arises in the initialization already.
{{ message }}
When using very large reservoirs (>5000) in the ESN, jax allocates too much memory and the code fails. The issue arises in the initialization already.