nnx.pmap documentation seems to be for vmap

Open hrbigelow opened this issue 1 year ago • 3 comments

Hi,

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

Oct 24 '24 21:10 hrbigelow

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

Oct 25 '24 09:10 cgarciae

No problem, thank you @cgarciae

Oct 25 '24 16:10 hrbigelow

will there be any updates soon...?

Aug 04 '25 05:08 HyperBlaze456