Techniques in optimizing neural network architectures