amatorch

Modules

constraints	Constraints to keep the filters on a certain set (e.g. sphere), or to fix some filter values during training.
data_wrangle	Utility functions for wrangling and subsampling data, outputs and statistics.
datasets	Load datasets and pre-trained filters.
inference	Compute probability of observed points given a set of distributions (and obtain the distribution parameters given a labeled dataset).
models	Subpackage with the implemented AMA model variants.
normalization	Functions to normalize tensors in different ways.
optim	Routine to fit AMA filters using Gradient Descent.
plot	Utilities for plotting model parameters (filters and statistics) and model outputs (responses, posteriors and estimates).
version