OpenReservoirComputing: GPU-Accelerated Reservoir Computing in JAX - v0.3.2

v0.3.2

JAX JIT Compatibility

All string-based fields (resdim, chunks, dtype, mode, etc.) across all ORC modules are now marked eqx.field(static=True). This makes ORC models fully compatible with jax.jit when the model is passed as an explicit pytree argument, and resolves ConcretizationTypeError failures in training functions like trainRCForecaster under jax.jit. A new JAX JIT Compatibility example notebook documents the supported patterns and sharp bits for jax.jit vs eqx.filterjit. The preference for eqx.filterjit is strongly advertised.

Type Checking

• ty has been added as a development dependency and integrated into the CI pipeline (ty check src/ runs on every push). Many incorrect type annotations across the codebase have been fixed as a result

Documentation

• Notebook outputs are now generated at docs build time via mkdocs-jupyter (execute: true), keeping notebook source files output-free in the repository.
• Fixed active navigation item color in dark mode.
• Fixed a broken figure reference in the JOSS paper (architecture_paper.pdf).

Other

• Removed implicit float64 enabling; users must now call jax.config.update("jaxenablex64", True) explicitly. A warning is raised if - float64 is not enabled at import time.
• Fixed bug with reversed order of resdim and instate for GRU driver.
• Removed plt.show() calls from built-in visualisation utilities.
• Significantly reduced repository clone size by rewriting git history to remove large historical blobs.
• Expanded test coverage for jit, vmap, and autodiff

- Python
Published by Jan-Williams about 2 months ago

OpenReservoirComputing: GPU-Accelerated Reservoir Computing in JAX - v0.2.0

Refactor of training functions to allow for train_RC... to work more easily with user instantiated models.

- Python
Published by Jan-Williams 5 months ago

OpenReservoirComputing: GPU-Accelerated Reservoir Computing in JAX - v0.1.3-alpha

- Python
Published by Jan-Williams 5 months ago

OpenReservoirComputing: GPU-Accelerated Reservoir Computing in JAX - v0.1.2-alpha

- Python
Published by Jan-Williams 5 months ago