rockpool.training

Contains packages for assisting with NN training

Modules

rockpool.training.adversarial_jax

Functions to implement adversarial training approaches using Jax

rockpool.training.ctc_loss

Implementations of the CTC loss, in numpy, Jax and Torch

rockpool.training.jax_debug

Utilities for debugging Jax training loops

rockpool.training.jax_loss

Jax functions useful for training networks using Jax Modules.

rockpool.training.torch_loss

Torch loss functions and regularizers useful for training networks using Torch Modules.