Lightning NF for JetNet [1].
Model uses mass constrained and conditioned flows for the generation of jets
Everything is logged to tensorboard
To set up the environment run pip install -r requirements.txt
To retrain Models to reproduce ACAT paper: use main.py
To reproduce results in paper use IML.ipynb
[1] R. Kansal et al. Particle Cloud Generation with Message Passing Generative Adversarial Networks NeurIPS 2021 (https://arxiv.org/abs/2106.11535)[2106.11535]