medium10 pts · first answer countsdeep-learningdistributed-training
How tensor parallelism splits a layer
In Megatron-style tensor parallelism across 4 GPUs, how is a large feed-forward layer handled?
Discussion 0
Join free to post in the discussion.
Related questions
- Parameters in nn.Linear(10, 5)Machine Learning Engineer
- Why sigmoid slows deep trainingMachine Learning Engineer
- Range of tanhMachine Learning Engineer
- Cross-entropy loss by handMachine Learning Engineer
- Telling autograd to track a tensorMachine Learning Engineer
- Why wrap inference in no_gradMachine Learning Engineer
- Logging loss without leaking memoryMachine Learning Engineer
- Feeding a single image to a CNNMachine Learning Engineer