Skip to content
This repository has been archived by the owner on Aug 11, 2022. It is now read-only.

Commit

Permalink
Update readme.md
Browse files Browse the repository at this point in the history
  • Loading branch information
@vpj
vpj committed Apr 11, 2022
1 parent ac43bfd commit b9b8490
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions readme.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@ We have taken out several optimizations in GPT-NeoX for simplicity.

### [Annotated implementation](https://lit.labml.ai/github/labmlai/neox/tree/main/src/neox/__init__.py)

[![Screenshot](https://github.com/labmlai/neox/blob/main/assets/annotated_gpt_neox_model.png)](https://lit.labml.ai/github/labmlai/neox/tree/main/src/neox/__init__.py)

### Sample usages

* [Notebook to download checkpoints and test the model](https://github.com/labmlai/neox/tree/main/notebooks/download_and_evaluate.ipynb)
Expand Down

0 comments on commit b9b8490

Please sign in to comment.