AIOS machine learning integration