Implement LightningModule & LoggerCallback in GraphGym

[aniketmaurya]

Partially address 2nd task from the roadmap #4503.

Lightning Module implementation for model training/val/test step and implement Logger as PL Callback.

[codecov]

Codecov Report

Merging #4531 (f09f687) into master (6fd6f5b) will increase coverage by 0.03%.
The diff coverage is 92.64%.

❗ Current head f09f687 differs from pull request most recent head 27afa37. Consider uploading reports for the commit 27afa37 to get more accurate results

@@            Coverage Diff             @@
##           master    #4531      +/-   ##
==========================================
+ Coverage   82.88%   82.92%   +0.03%     
==========================================
  Files         316      316              
  Lines       16679    16747      +68     
==========================================
+ Hits        13825    13888      +63     
- Misses       2854     2859       +5     

Impacted Files	Coverage Δ
torch_geometric/graphgym/logger.py	81.95% <89.36%> (+2.36%)	⬆️
torch_geometric/graphgym/model_builder.py	98.18% <100.00%> (+1.12%)	⬆️

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 6fd6f5b...27afa37. Read the comment docs.

[rusty1s]

Thank you!

[rusty1s]

Thank you! Looking forward to the follow-up contributions :)