vit-pytorch state of the art image classifier with multiple attention heads
feed forward neural network