TMVA Deep Learning Developments - 3D Convolutions for GPU


Toolkit for Multivariate Analysis (TMVA) is a multi-purpose machine learning toolkit integrated into the ROOT scientific software framework, used in many particle physics data analysis and applications. We have expanded TMVA’s capabilities to include a deep learning library (DNN) supporting 2Dconvolutional layers with training on both CPU and GPUs. This summer we would like to complete it with the 3D convolution operator. 3D CNNs have very promising applications in particle physics, such as imaging calorimetry and particle tracking, allowing physicists to use new techniques to identify particles and search for new physics. In addition they can be used as Generative Adversarial Networks for fast imaging detector simulations.

Task ideas and expected results

The initial task of the student will be validating and integrating in ROOT the CPU implementation for 3D convolution that was developed last summer . Afterwards, the main task will be the development of the 3D convolution for Nvidia GPU using the cuDNN library, following the current implementation present in ROOT/TMVA for the 2D operator. The developed code should support both the training and inference for deep learning models.


Strong C++ skills, solid knowledge of deep learning, familiarity with GPUs


Please DO NOT contact mentors directly by email, and instead please send project inquiries to with Project Title in the subject and relevant mentors will get in touch with you.

Corresponding Project

Participating Organizations