We have recently developed a neural-network representation of the quantum mechanical wave function of light nuclei. To extend our approach to large nuclei, we need to leverage GPUs. The JLSE testbed machines are ideally suited to carry out this development work. So far, we have been extensively using Pytorch, but we have now switched to Tensorflow and Google JAX. The long-term goal is to utilize Aurora efficiently.