GitHub - lutzroeder/models: A minimal version of GPT-2 using PyTorch and Gemma 3 using JAX. · GitHub
Skip to content

lutzroeder/models

BranchesTags

Folders and files

Repository files navigation

Models

gemma3 - A minimal implementation of Gemma 3 in 200 lines only depending on JAX and SentencePiece.

gpt2 - A minimal implementation of GPT-2 in 175 lines only depending on PyTorch.

About

A minimal version of GPT-2 using PyTorch and Gemma 3 using JAX.

Resources

Readme
Activity

Stars

47 stars

Watchers

1 watching

Forks

4 forks
Report repository

Releases

No releases published

Packages

 
 
 

Contributors

Languages