It was added recently and just added to a new release, so pip install transformer_lens should work now/​soon (you want v1.16.0 I think), otherwise you can install from the Github repo
pip install transformer_lens
It was added recently and just added to a new release, so
pip install transformer_lens
should work now/​soon (you want v1.16.0 I think), otherwise you can install from the Github repo