Using the DNN from the previous milestone, we should add features so that a model replica can be further divided into smaller units in case a model doesn't fit in a single machine.