python machine-learning deep-learning transformers pytorch neural-networks attention-mechanism sparse-autoencoders model-analysis interpretability ai-research mixture-of-experts mechanistic-interpretability model-surgery model-composition