Implementations of some models in JAX.
| Model | Architecture Type | Reference |
|---|---|---|
| GPT-2 | Decoder-only Transformer (causal LM) | ref |
More to come.
-
Notifications
You must be signed in to change notification settings - Fork 0
VascoSch92/models
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Implementations of some models in JAX
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published