[P0] Fix subspace broadcast issue

stanfordnlp / pyvene

Stanford NLP Python Library for Understanding and Improving PyTorch Models via Interventions

http://pyvene.ai

Apache License 2.0

652 stars 63 forks source link

Open PinetreePantry opened 3 months ago

PinetreePantry commented 3 months ago

Subspace broadcasting did not handle many cases that require broadcasting. This has been causing subtle bugs in pyvene calls with subspaces