neelnanda-io/TransformerLens

This is stars and forks stats for /neelnanda-io/TransformerLens repository. As of 25 Apr, 2024 this repository has 484 stars and 120 forks.

TransformerLens This library is maintained by Joseph Bloom and was created by Neel Nanda Read the Docs Here Installation Install: pip install transformer_lens import transformer_lens # Load a model (eg GPT-2 Small) model = transformer_lens.HookedTransformer.from_pretrained("gpt2-small") # Run the model and get logits and activations logits, activations = model.run_with_cache("Hello World") Key Tutorials Introduction to the Library and Mech Interp Demo of Main TransformerLens Features A Library for...