talwarabhimanyu/tensor_parallel_toy_model

Tensor parallelism demo using a toy model trained on a regression task.

Python