Implement BroadcastAxes::vmap#3319

Aristide021 wants to merge 4 commits intoml-explore:mainml-explore/mlx:mainfrom

Aristide021:broadcastaxes-vmapAristide021/mlx:broadcastaxes-vmapCopy head branch name to clipboard