The next release of declearn should incorporate the following features:

• GPU support with explicit device-placement policy.
• Framework-specific optimizers support using OptiModule wrappers.
• Better support for frozen neural network weights (fix bugs).
• R² evaluation metric for regression problems.

The following features may or may not be part of the release depending on timing:

• Better testing of OptiModule state-access and support for RNG state access.
• Better support for tf.IndexedSlices row-sparse gradients.
• New functional tests for model classes using toy datasets.