v1.3.1

Miscellaneous fixes.

• tl.Embedding now has the same signature as pytorch/tf
• train.lr_schedule (function object) -> train.lr_schedule_fn (function)
• Report loss back to training.Loop