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.