GitHub repository: lucidrains/PaLM-jax
Primary language: Python
Created: 2022-04-08T15:34:36Z
Description: Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)