TMS-Kit

Toolkit for running toy models of superposition experiments in PyTorch.

Installation

Releases are hosted on PyPI, and can be installed using pip as below:

pip install tms-kit

Paper Reproductions

This library provides utilities for reproducing experiments on toy models of superposition as described in the following papers:

• Toy Models of Superposition

• Toy Models of Feature Absorption

See the reproductions section for more details.

Contents

• Basic Usage
  • Minimal Example
• Paper Reproductions

Project Links

• GitHub
• PyPI