=== A More Intermediate Example ===

Here is a toy model that contains two linear layers. Each linear layer is designed to run a separate GPU.