PaLM-jax
Python
★ 189
updated 4y ago
Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
No plain-English explanation yet — one is being written right now. Check back in a minute.