Simple Annotated implementation of GPT-NeoX in PyTorch
Home Page:https://lit.labml.ai/github/labmlai/neox/tree/main/src/neox/__init__.py
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool