Tianyu He
Results
2
comments of
Tianyu He
I am also curious about this. Having a pmap similar to the one in jax would be very handy for customizing functions.
same problem