A miniture AI training framework for PyTorch - View it on GitHub