In the code here: https://www.kaggle.com/ryanholbrook/detecting-the-higgs-boson-with-tpus
Before the model is compiled, the model is made using this code: