Neural network layers

Feature extraction layers

Spectrogram(*args, **kwargs)	A module that computes a spectrogram from a waveform using JAX.
LogMelFilterBank(*args, **kwargs)	A module that converts spectrograms to (log) mel spectrograms.
MFCC(*args, **kwargs)	A module that computes Mel-Frequency Cepstral Coefficients (MFCCs).

Data augmentation layers

DropStripes(*args, **kwargs)	A module that randomly drops stripes (time or frequency bands) from a spectrogram.
SpecAugmentation(*args, **kwargs)	A module that applies SpecAugment data augmentation to spectrograms.