Work Uni Vision Transformer Implementation of the Vision Transformer Architecture in PyTorch Fun Auto Compyute Lightweight Autograd Engine in Pure Python