MegatronLM: Model Parallelism Using Multiple Tensors in PyTorch

Saliya Ekanayake
Aug 13 · 1 min read
Saliya Ekanayake

Written by

Postdoctoral Fellow, Berkeley Lab, CA

Welcome to a place where words matter. On Medium, smart voices and original ideas take center stage - with no ads in sight. Watch
Follow all the topics you care about, and we’ll deliver the best stories for you to your homepage and inbox. Explore
Get unlimited access to the best stories on Medium — and support writers while you’re at it. Just $5/month. Upgrade