v0.2.1

v0.2.1 (2024-07-02)

Fix

• fix: attribution calculation (#14)

• add test for pruner

• move walkthrough to top-level directory

• ignore pickle files

• test caches match TL caches

• add motivation for cache dict

• fix viz

• delete incorrect test

We will produce one hook for every hook point in the model which needs to be cached
This is different from the number of src nodes in the graph

• rename file to avoid pytest issue

---

Co-authored-by: Daniel CH Tan <dtch1997@users.noreply.github.com> (8dc8e5a)

Unknown

• improve docs, readme (8bbd72e)

• refactor nodes (f4fb9f1)

v0.2.0

v0.2.0 (2024-06-27)

Chore

• chore: mark test as xfail (2a9a15f)

Feature

• feat: pruning, evaluation (#11)

• wip: single-batch debugging

• single-batch in original notebook

• fix: scores not added inside loop

• wip add pruning algorithms

• fix formatting

• delete pickle files

• rafactor

• refactor

• refactor

• wip walkthrough

---

Co-authored-by: Daniel Tan <dtch1997@users.noreply.github.com> (51d8223)

Refactor

• refactor: attribute (397b366)

• refactor: var names (a7e1904)

Unknown

• add code to export attrib scores in example (fc84984)

• mwe for run_attribution function (34943c1)

v0.1.1

v0.1.1 (2024-06-24)

Fix

• fix: attentionode get_act (8f69328)

Refactor

• refactor: separate model and node cache

fix: tests, types (ae8b6be)

• refactor: node (#4)

• refactor graph to generic

• refactor build

• minor

• refactor index

• add data handler

• attribute tests passing

• fix tests

---

Co-authored-by: Daniel Tan <dtch1997@users.noreply.github.com> (44b89c7)

v0.1.0

v0.1.0 (2024-06-19)

Feature

• feat: attribute (#2)

• add unit tests for graph

• minor

• WIP: add notebook to do attribution

• write test for graph indexing

• working tests for indexer

• refactor indexer: names, method names

• implement attribution function

• fix types

---

Co-authored-by: Daniel Tan <dtch1997@users.noreply.github.com> (623412d)

Unknown

• refactor to MultiDiGraph (8c479d3)

v0.0.0

v0.0.0 (2024-06-15)

Chore

• chore: disable pypi ci (6ed0e1e)

• chore: update README (6b79576)

Refactor

• refactor: graph (#1)

• refactor graph

• Add minimal node, edge dataclasses

• implement minimal graph

• working build_graph

• setup graphviz in github actions

• remove py312 which has no tensordict

• fix python req

• move eap code out of library

• remove unused deps

---

Co-authored-by: Daniel Tan <dtch1997@users.noreply.github.com> (890228b)

Unknown

• minor (e83a1a6)

• EAP working (ee6d7cb)

• WIP add old SAEs (5ceb1a0)

• initial commit (92ebb7b)