MULTI-GPU TRAINING AND PARALLEL CPU COMPUTING FOR THE MACHINE LEARNING EXPERIMENTS USING ARIADNE LIBRARY

Modern machine learning (ML) tasks and neural network (NN) architectures require huge amounts ofGPU computational facilities and demand high CPU parallelization for data preprocessing. At thesame time, the Ariadne library, which aims to solve complex high-energy physics tracking tasks withthe help of deep neural networks, lacks multi-GPU training and efficient parallel data preprocessing onthe CPU.In our work, we present our approach for the Multi-GPU training in the Ariadne library. We willpresent efficient data-caching, parallel CPU data preprocessing, generic ML experiment setup forprototyping, training, and inference deep neural network models. Results in terms of speed-up andperformance for the existing neural network approaches are presented with the help of GOVORUNcomputing resources.