@Mubsi
https://fixexception.com/jax/unbound-axis-name-axis-name-the-following-axis-names-e-g-defined-by-pmap-are-available-to-collective-operations-named-axes/
Lab ID: uibbbhks
Hi Abhijit1,
I presume this has been resolved. If not, please open a new thread.