Distributed systems for neural network models