Basic all_gather implementation#1663

Chapaman wants to merge 2 commits intoelixir-nx:mainelixir-nx/nx:mainfrom

Chapaman:all_gather-implementationChapaman/nx:all_gather-implementationCopy head branch name to clipboard