search

stanfordnlp / pyvene

Stanford NLP Python Library for Understanding and Improving PyTorch Models via Interventions
http://pyvene.ai
Apache License 2.0
608 stars 59 forks source link

Add preliminary scatter() function tests #82

Closed ZhengPeterWang closed 8 months ago

ZhengPeterWang commented 8 months ago

First scatter_neurons() function tests. Will add more tests covering all cases of scatter() as that is more complicated than gather_neurons()

frankaging commented 8 months ago

Thanks! LGTM and Ship It!