Bayes Optimal Learning of Attention-Indexed Models Code repository for "Bayes optimal learning of attention-indexed models"