chiyuri / palm-jax Goto Github PK
View Code? Open in Web Editor NEWThis project forked from lucidrains/palm-jax
Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
License: MIT License