Researchers must often choose between Bayesian or neural network models of behavior, two paradigms with complementary strengths and weaknesses. An ideal paradigm would facilitate testing many kinds of ...

The main scripts automatically choose CUDA when available, then Apple Metal (MPS), and otherwise CPU, through train_utils.get_best_device(). Training the approximately 30-million-parameter networks ...