[ml][single] Implemented communication in ML API