v2.2.0

What's Changed

• Update onnx export and validation for cms by @erwulff in #378
• Unify key4hep by @farakiko in #374
• allow missing layers in models when loading checkpoint by @kawaho in #380
• Plots for AN V1 by @jpata in #381
• Refactor training code by @jpata in #384
• Compute and track detailed evaluation metrics on each epoch by @jpata in #385
• Use per-particle pt weight in loss by @jpata in #383

New Contributors

• @kawaho made their first contribution in #380

Full Changelog: v2.1.0...v2.2.0