prob contributions by individual neuron rather than layer

[nathanneuro]

working on some code to expand your causal trace demo to look at the prob contributions by individual neuron rather than layer. It's pretty messy still but if you're interested I'd be happy to share it.

[kmeng01]

this is super cool! we'll take a look at the PR in a few days once things clear up — thanks for the contrib :)

[caiyuchen-ustc]

Hello, nathanneuro! With the same idea as you, I was wondering if you could share some more details about your project and its outcomes. I'm particularly interested in the specific results and any finer details you could provide. I appreciate your time and thank you in advance!