I think we used a distill.pub template. Also Jerry wrote some custom BNN fitting code in javascript. I'll ask my co-authors to open-source it.
Update: the code is here:
https://github.com/jerryqhyu/distill_bayes_net