functorch icon indicating copy to clipboard operation
functorch copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Will pmap be supported in functorh？

Open shixun404 opened this issue 2 years ago • 2 comments

Greetings, I am very grateful that vmap is supported in functorch. Is there any plan to include support for pmap in the future? Thank you. Additionally, what are the ways that I can contribute to the development of this project?

shixun404 avatar Jan 11 '23 17:01 shixun404

Any updates on this?

jordiae avatar Jun 26 '23 19:06 jordiae

I am also curious about this. Having a pmap similar to the one in jax would be very handy for customizing functions.

ablghtianyi avatar Jun 05 '24 16:06 ablghtianyi