Add register tokens
https://arxiv.org/abs/2309.16588
As discussed in todays meeting: https://indico.cern.ch/event/1362253/contributions/5761535/
Register tokens add slight improvement in performance (up to 10%). This was tested on 25M jets and 25 epochs.
Edited by Samuel Van Stroud