A JAX research toolkit for building, editing, and visualizing neural networks.

The nnsight package enables interpreting and manipulating the internals of deep learned models.

Efficiently computing & storing token n-grams from large corpora

TransformerLens + HuggingFace

A markup-based typesetting system that is powerful and easy to learn.

Code and documentation to train Stanford's Alpaca models, and generate the data.

Port of OpenAI's Whisper model in C/C++