transformers pytorch jax safetensors gpt2 text-generation question-generation dataset:squad arxiv:1606.05250 arxiv:1705.00106 autotrain_compatible text-generation-inference endpoints_compatible region:us