Augmenting neural networks

Abstract

Methods, systems, and apparatus, including computer programs encoded on computer storage media, for augmenting a neural network with additional operations. One of the methods includes maintaining, by a computational graph system that manages execution of computational graphs representing neural network operations for users of the computational graph system, data specifying a plurality of pre-trained neural networks, wherein each of the pre-trained neural networks is a neural network that has been trained on training data to determine trained values of the respective parameters of the neural network; obtaining data specifying a user computational graph representing neural network operations, the user computational graph comprising a plurality of nodes connected by edges; identifying (i) an insertion point after a first node in the user computational graph and (ii) a particular pre-trained neural network from the plurality of pre-trained neural networks; and inserting a remote call node into the user computational graph.