Support for covariance matrices

yliess86 / BayeFormers

General API for Deep Bayesian Variational Inference by Backpropagation. The repository has been designed to work with Transformers like architectures. Compatible with the HuggingFace Transformers models.

MIT License

41 stars 3 forks source link

Support for covariance matrices #2

Open sgbaird opened 2 years ago

sgbaird commented 2 years ago

Hi, does BayesFormer support output covariance matrices? (i.e. covariances between inputs)