Implementation of the BayesFormer on a simple VIT (https://github.com/lucidrains/vit-pytorch) by the great lucidrains
@misc{2206.00826,
Author = {Karthik Abinav Sankararaman and Sinong Wang and Han Fang},
Title = {BayesFormer: Transformer with Uncertainty Estimation},
Year = {2022},
Eprint = {arXiv:2206.00826},
}