Add W&B experiment tracking and structured logging

Add training loop, colab notebook, and final README

before hf space

intial commit