timaeus-research / devinterp

Tools for studying developmental interpretability in neural networks.
77 stars 14 forks source link

Add histograms for arbitrary traces of ChainCallbacks #43

Closed georgeyw closed 1 year ago

georgeyw commented 1 year ago

Main intent is to be able to use this for losses with OnlineLLCEstimator, but this might be helpful for other metrics we compute as well