Wes Gurnee's starred repositories
numerical-linear-algebra
Free online textbook of Jupyter notebooks for fast.ai Computational Linear Algebra course
Mathematics-for-ML
🧮 A collection of resources to learn mathematics for machine learning
TransformerLens
A library for mechanistic interpretability of GPT-style language models
lovely-tensors
Tensors, for human consumption
git-re-basin
Code release for "Git Re-Basin: Merging Models modulo Permutation Symmetries"
awesome-neuro-ai-papers
Papers from the intersection of deep learning and neuroscience
PyTorch-Model-Compare
Compare neural networks by their feature similarity
blackbox-backprop
Torch modules that wrap blackbox combinatorial solvers according to the method presented in "Differentiating Blackbox Combinatorial Solvers"
jax_verify
Neural network verification in JAX
glasgow-subgraph-solver
A solver for subgraph isomorphism problems, based upon a series of papers by subsets of McCreesh, Prosser, and Trimble.
interpretability-starter
🧠 Starter templates for doing interpretability research
transformer-utils
Utilities for the HuggingFace transformers library
activation_additions
Algebraic value editing in pretrained language models
state-probes
Code for the paper "Implicit Representations of Meaning in Neural Language Models"
spacy-alignments
💫 A spaCy package for Yohei Tamura's Rust tokenizations library
allfed-integrated-model
Integrated model to calculate the effects of resilient foods in catastrophic events
llm-context-neurons
Find context neurons in Pythia models.
swap-graphs
An implementation of input swap graphs. A tool to discover the role of neural network components with causal interventions.
fairmandering_paper
Code repository for Fairmandering: A column generation heuristic for fairness-optimized political districting