Graph Neural Network Tracking

This open source project uses graph neural networks to reconstruct charged particle trajectories in particle detectors ("tracking"). Batteries are included: The package implements the whole processing pipeline, several models and approaches, and the evaluation of the final performance metrics. Built around pytorch lightning, the models are easy to train and to restore. By using hooks and callbacks, everything remains modular and maintainable.