/neox

Simple Annotated implementation of GPT-NeoX in PyTorch

Primary LanguagePythonMIT LicenseMIT

Watchers