shakespeare-lm

August 18, 2025 ยท View on GitHub

Shakespeare-lm is an extremely small model that can generate Shakespeare-style text given a prompt, written in Catniff. It is a GRU (gated recurrent unit) behind the hood and uses a character-level tokenizer.

Setup

  1. Clone this repo
  2. Install Catniff:
npm install

Training and inference

Have a look at ./index.js, there is already code that is commented for training and inference, simply uncomment and run.