Pytorch Lightning implementation of Vision Transformer with support for loading checkpoints saved in official Flax implementation.