Decompose small kernel-depth 3D convs into 2D convs (#3625)#3785

katlun-lgtm wants to merge 4 commits into

ml-explore:mainml-explore/mlx:mainfrom

katlun-lgtm:conv-3d-small-kd-decompkatlun-lgtm/mlx:conv-3d-small-kd-decompCopy head branch name to clipboard