Machine learning toolkit