Pytorch implementation of the HLS4ML 3 Layer Jet Tagging model, including a standard Pytorch (float) and a Quantized (via Xilinx's Brevitas library) implementation. designed for use with the HLS4ML LHC Jet dataset (100 particles): https://zenodo.org/record/3602254
weight_dists
directoryNote this repo is also hosted on GitLab: https://gitlab.nautilus.optiputer.net/jmduarte/pytorch-jet-classify and the corresponding Docker container is hosted there too: https://gitlab.nautilus.optiputer.net/jmduarte/pytorch-jet-classify/container_registry