Common Utilities

Datasets

WavHDF5Dataset(dataset_path[, duration, sr, ...])

Dataset class for loading and preprocessing multi-channel audio data stored in HDF5 format.

Network Building Blocks

ResSequential()

Probabilistic Distributions

ApproxBernoulli(temperature[, probs, ...])	Approximation of Bernoulli distribution using a relaxed Bernoulli formulation.
ComplexWishart(nu, covariance_matrix)	Complex Wishart distribution.
ComplexInverseWishart(nu, covariance_matrix)	Complex Inverse Wishart distribution.
FastComplexWishart(nu, covariance_matrix)	Fast approximation of Complex Wishart distribution.
FastComplexInverseWishart(nu, covariance_matrix)	Fast approximation of Complex Inverse Wishart distribution.

SCM_estimators

EmEstimator(n_iter[, eps])

Expectation-Maximization (EM) algorithm for estimating spatial covariance matrices (SCMs).

Diagonalizers

ISSDiagonalizer([n_iter, eps, eps2, eps3, ...])

Iterative Source Steering (ISS) algorithm for diagonalizing spatial covariance matrices.

Lightning Callbacks

ConstantAnnealerCallback(name, value, period)	Lightning callback that temporarily overrides a module attribute during training.
CyclicAnnealerCallback(name, cycle, max_value)	Lightning callback that cyclically varies a module attribute during training.
CyclicStopAnnealerCallback(name, cycle, ...)	Lightning callback that cyclically increases a module attribute and stops after a period.