Heterogeneity, hyperparameters, and GPUS: towards useful transport calculations using neural networks

We use a neural network to solve the discrete ordinates neutron transport equation in slab geometry. We presented the solution to six different slab geometry neutron transport problems using neural networks. We extend previous work which solved homogeneous medium problems by considering problems with spatial heterogeneity. We also include a hyper-parameter study and a description of our GPU implementation along with the runtime we observed using Sierra GPUs at LLNL. (authors)