Nested pmap() functions don not work as I expect

Nested pmap seems to work fine. See my notebook for an example.