Python-based backend for machine learning tasks