Illustration drawn by RepoMasters from the repo’s ID and its language color. Credits
PrototypeCommon

lucidrains/PaLM-jax

Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)

Owner
lucidrains (personal account)
Stars
Stars: 189
Forks
Forks: 10
Language
Python
License
MIT
Created
2022
Topics
deep-learningtransformersattention-mechanismartificial-general-intelligencejax
Tier
Prototype (100 to 249 ★) What this tier means
View on GitHub

More repos by lucidrains

All 175 repos by lucidrains

Nearby in Python

All Python repos