transformers pytorch tf jax vit image-classification vision dataset:imagenet dataset:imagenet-21k arxiv:2010.11929 arxiv:2006.03677 license:apache-2.0 autotrain_compatible endpoints_compatible region:us