torchtitan
torchtitan copied to clipboard
add pipeline parallelism to train llama
I will start a sep branch for pippy integration and begin testing with this.
3D parallel enabled in #344