[LuxTraining] Wrappers for less clunky training loops

[avik-pal]

TODOs:

• Tests
• Tutorial
• Detailed docs on what the loss function looks like

Non-goals:

• Something like FluxTraining (or FastAI)

[codecov]

Codecov Report

Merging #104 (c97a83a) into main (c97a83a) will not change coverage.
The diff coverage is n/a.

❗ Current head c97a83a differs from pull request most recent head 783a2b0. Consider uploading reports for the commit 783a2b0 to get more accurate results

@@           Coverage Diff           @@
##             main     #104   +/-   ##
=======================================
  Coverage   76.11%   76.11%           
=======================================
  Files          14       14           
  Lines        1030     1030           
=======================================
  Hits          784      784           
  Misses        246      246           

---

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 c97a83a...783a2b0. Read the comment docs.