What is jsGPT?

jsGPT is a project that seeks to implement various model architectures in TensorflowJS, most notably NanoGPT. By implementing small, easy-to-understand, and easily trainable architectures, jsGPT is especially useful to beginners hoping to understand the basics of modern machine learning.

What is the jsGPT Lab?

This website, also known as the jsGPT Laboratory, is a place where users can easily experiment with the models that have so far been implemented in jsGPT. Simply tune the hyperparameters to your liking, and see the model train in real time, with outputs for you to analyze once training is complete.

How does this site work? Is there a backend?

Nope! This is a fully frontend project, made possible by TensorflowJS. By using TensorflowJS, you train models on your own machine. Leveraging the power of WebGPU, jsGPT is able to train models quickly entirely without a backend! This does mean you are responsible if you try training a model with some crazy hyperparameters, though. Don't fry your GPU please.

Can I see the source code?

Of course, it's right here! Please star the repository if you like my work, and feel free to refer me to your tech company's summer internship.

back to home