#116 · Primary category: Foundation Models

PaLM-pytorch

artificial-general-intelligence attention-mechanism deep-learning transformers

Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways

Project last updated:11/09/22

GitHub Stars

824

Forks

81

Contributors

3

License

MIT

Why we included this project

PaLM-pytorch is a compact PyTorch port of the transformer architecture Google described in its PaLM paper, and the whole thing fits in under 200 lines. That brevity is deliberate: the author wrote it as a teaching tool, not a production model, so you can trace the attention mechanism, feed-forward blocks, and normalization choices straight from the paper without digging through a large codebase. You can spin up a small PaLM-style model, train it on something like Enwik8, and see the design decisions play out. It won't scale to 540B parameters, but for anyone learning transformer internals or borrowing PaLM-style ideas for their own experiments, the clean, dependency-light code is easy to adapt.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category