Now we just need an iterative solver over both the structure and the "weights", and we get both architecture search and training at the same time