Integration with ML Frameworks