Abstract: One embodiment of the present invention sets forth a technique for managing machine learning. The technique includes organizing a set of reusable components for performing machine learning under a framework. The technique also includes representing, within the framework, a machine learning model as a graph-based structure that includes nodes representing a subset of the reusable components and edges representing input-output relationships between pairs of the nodes. The technique further includes validating the machine learning model based on inputs and outputs associated with the nodes and the input-output relationships represented by the edges in the graph-based structure. Finally, the technique includes generating the machine learning model according to the graph-based structure and configurations for the subset of the reusable components.