Machine learning frameworks in Python