Tensor Parallelism with JAX + Shard Map
Primary LanguagePythonMIT LicenseMIT
No issues in this repository yet.