top of page

Unique AI Framework

- (UAIF) -

CoE RAISE follows the rules of open science and publishes its results open-access when they are ready for wider application. All developments of CoE RAISE are being integrated into the Unique AI Framework (UAIF), which will not only contain the trained models but also documentation on how to use them on current Petaflop and future Exascale HPC, prototype, and disruptive systems. The developments toward the Unique AI Framework are continuously progressing. The present code base of the UAIF can be found below.

Machine-Learned Particle-Flow (MLPF)


Machine-Learned Particle-Flow (MLPF) is an algorithm based on Graph Neural Networks (GNN) and is aimed at performing efficient, GPU-accelerated particle flow reconstruction at large particle detector experiments. It takes particle tracks and calorimeter clusters as input and gives higher-level physics objects, for instance electrons, hadrons and photons, as output. This repository contains the code necessary to train MLPF using single or multiple GPUs, to perform large-scale hyperparameter optimization (HPO) using multiple compute nodes on HPC systems, to evaluate the model performance as well as to export the model for later use in inference. The main model and training is implemented in TensorFlow while Ray Tune is used for HPO. A publicly available dataset is available at [1]. MLPF was first introduced in [2] and later versions appeared in [3,4].

AI for HPC


AI4HPC, part of CoE RAISE, is an open-source library to train AI models with CFD datasets on HPC systems. In CoE RAISE, innovative AI methods on heterogeneous HPC architectures capable of scaling towards Exascale are developed and generalized for selected representative simulation codes and data-driven workflows. AI4HPC consists of data manipulation routines tuned to handle CFD datasets, ML models useful for CFD analyses, and optimizations for HPC systems. AI4HPC also includes a benchmarking suite to test the limits of any system with CPUs and GPUs towards Exascale and a HyperParameter Optimization (HPO) suite for scalable HPO tasks. 

AI4Sim Model Collection

This repository proposes convolutional (CNN) and graph-based neural network (GNN) architectures as physical surrogates in various industrial use cases. While CNNs can reach state-of-the-art performances on structured grids, GNNs offer natural surrogates for simulations relying on complex unstructured meshes. Several use cases are presented, with comparisons of training pipelines based on CNNs and on GNNs.



PhyDLL (Physics Deep Learning coupLer) is an open-source coupling library ( It allows a performant data transfer and processing between massively parallel physical solvers and distributed deep learning inferences. PhyDLL proposes different coupling schemes that suit the context and the data-structure topology. Currently, Fortran and Python interfaces are available for physical solvers and deep learning engines respectively. The ongoing collaborations within CoE RAISE, will enable the creation of a C/C++ interface in order to making PhyDLL even more accessible for a wider range of users.  Toward exascale, the development of inter-GPU communications in multi-nodes settings within PhyDLL is under way.  


[1]  J. Pata, J. Duarte, J.R. Vlimant, M. Pierini, M. Spiropulu "Simulated particle-level events of ttbar and QCD with PU200 using    

     Pythia8+Delphes3 for machine learned particle flow (MLPF)”

[2] J. Pata, J. Duarte, J.-R. Vlimant, M. Pierini, and M. Spiropulu, “MLPF: Efficient machine-learned particle-flow reconstruction using graph

      neural networks,” Eur. Phys. J. C, vol. 81, no. 5, p. 381, 2021.

[3] J. Pata et al. “Machine Learning for Particle Flow Reconstruction at CMS” 

[4] E. Wulff et al. “Hyperparameter optimization of data-driven AI models on HPC systems" 

[5] Baldi, P. (2012). Autoencoders, Unsupervised Learning, and Deep Architectures. Proceedings of ICML Workshop on Unsupervised and Transfer Learning, Proceedings of Machine Learning Research, 27, 37–49.

bottom of page