lucidrains/PaLM-jax

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

语言：Python · GitHub 星标：189 · Fork：10

话题：artificial-general-intelligence · attention-mechanism · deep-learning · jax · transformers

在 GitHub 上查看原仓库