ViT

[egozi]

add a few things:

1. default parameters for depth, heads etc.
2. standard models like - vit_base, vit_large, …
3. the interpolation module for diff image sizes.
4. What about pre trained weights? and an option flag.

Originally posted by @egozi in #200 (comment)