GPT-2 from scratch with torch