jsGPT Lab

Experiment with a simple bigram model, implemented in TensorflowJS!

Tune the hyperparameters below to your liking and hit "train" to train our bigram model on Shakespeare's The Tragedy of Coriolanus. You'll see model outputs improve slightly to mimic the Shakespearean writing style.


Note: please use a browser that supports WebGPU, WebGL, or WASM. This model runs on your own machine, NOT a backend — don't go too crazy with the hyperparameters.


Inputs


Hyperparameters

The bigram model's architecture is predetermined by the tokenization scheme, so the following hyperparameters only impact the model's training.







Outputs



Confused about the hyperparameters?

Want to try a more complex model?

What is jsGPT?