Exercise in `vmap` notebook on message passing is actually an example

[ericmjl]

As per title. Either decide to set it as an exercise or set it as an example.