Tensor Parallelism with Jax.pjit Posted by irhshafkat on Monday October 10, 2022 @09:24PM from the irhum.github.io dept. https://irhum.github.io/blog/pjit/ ( 1 comment | on HN )