Hey team, love the project! I noticed that the training scripts inside train/sft/ currently lack type hints, which makes debugging raw data and configuration dicts a bit tricky.

I'd love to contribute by progressively adding strict type annotations to these files to improve maintainability and catch potential bugs early. Would you welcome a PR for this? I can start with a couple of the core utility scripts first