graykode/gpt-2-Pytorch

Health v2 · maintenance only

Flags: no_releases

Adoption not part of the score

1013 stars · 230 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

A compact PyTorch reimplementation of OpenAI's GPT-2 for text generation, loading Hugging Face's converted pre-trained weights. It provides a simple script for generating text continuations from a prompt.

Use cases

• generate text continuations from a prompt with GPT-2 in PyTorch
• study a minimal readable GPT-2 implementation
• run GPT-2 inference without TensorFlow
• experiment with transformer language models
• generate stories from seed text

When to choose

• you want a small, readable GPT-2 codebase in PyTorch for learning or experimentation
• you need lightweight GPT-2 text generation without the original TensorFlow stack

When to avoid

• you need modern LLMs, fine-tuning pipelines, or maintained tooling - use Hugging Face Transformers instead
• you need production-grade inference or recent model support

Facets

library · maturity abandoned

llm-inference nlp machine-learning large-language-models deep-learning python cross-platform gpt-2 pytorch text-generation transformer language-model natural-language-processing

Member repositories

For agents

markdown · JSON · MCP: product_card(name="graykode/gpt-2-Pytorch")