a small code base for training large models - View it on GitHub