Machine learning framework