nnx.pmap documentation seems to be for vmap

[hrbigelow]

Hi,

The source code documentation for nnx.pmap appears to be instead referring to nnx.vmap.

[cgarciae]

This is correct. Sorry it was adapted very quickly from vmap for an experiment.
Will update soon.

[hrbigelow]

No problem, thank you @cgarciae

[HyperBlaze456]

will there be any updates soon...?