transformers pytorch gpt_jx text-generation text-generation-inference custom_code en code dataset:tiiuae/falcon-refinedweb dataset:bigcode/the-stack-dedup dataset:cerebras/SlimPajama-627B arxiv:1706.03762 arxiv:2306.01116 arxiv:2002.05202 arxiv:2211.15533 license:mit autotrain_compatible region:us