Fix: routing gate visualization (replace logit lens with gate argmax) eda2c9d verified DuoNeural commited on Jun 12
