According to an aspect, there is disclosed a method comprising receiving, from a producer node, producer node capabilities comprising a machine learning based assistance capability, the machine learning based assistance capability comprising machine learning based functionalities, each machine learning based functionality comprising a machine learning entity and at least one machine learning mode associated with the machine learning entity. According to another aspect, there is disclosed a method comprising identifying producer node capabilities comprising a machine learning based assistance capability, the machine learning based assistance capability comprising machine learning based functionalities, each machine learning based functionality comprising a machine learning entity and at least one machine learning mode associated with the machine learning entity; and causing transmission of the producer node capabilities to a consumer node.