lucidrains/PaLM-jax
Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
PythonMIT
Issues
- 0
Equinox fails with deprecated message
#4 opened by jojivk73 - 0
Vmap is required
#3 opened by naveens01 - 0
question about vmap
#2 opened by yiyixuxu