4 ms·Tensor Parallelism with Jax.pjit2 points by irhshafkat 4y agoirhshafkat 4y agoAuthor here, felt it might be of interest to the folks on here!