Please also take a look at the ablation-study branch to find the notebooks to reproduce the results from the ablation study (and other tables) in the paper. Note how the paths through the channel ...