Fix vmap + floor_divide: preserve integer dtype#3292

zcbenz merged 4 commits intoml-explore:mainml-explore/mlx:mainfrom

robert-johansson:fix-vmap-floor-dividerobert-johansson/mlx:fix-vmap-floor-divideCopy head branch name to clipboard