This is in 'nitpick territory', but I think it would be more conventional to call name this file/module `neural_network_design`