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