pytorch jax gpt2 lm-head causal-lm transformers text-generation
.gitattributes README.md config.json flax_model.msgpack merges.txt optimizer.pt pytorch_model.bin scheduler.pt special_tokens_map.json tokenizer_config.json trainer_state.json training_args.bin vocab.json
Github 
- https://github.com/SIC98/GPT2-python-code-generator