timaeus-research / devinterp

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

Separate visualization-related dependencies into an extra #83

Open jqhoogland opened 4 months ago

jqhoogland commented 4 months ago

Some people might want to roll out their own visualization code. One thing that could make it easier to support this is to offload the visualization dependencies (matplotlib, pandas, ...) into an optional dependency, so you can install this with a flag like pip install devinterp[viz].