/einshard

Einsum-like high-level array sharding API for JAX

Primary LanguagePythonMIT LicenseMIT

Watchers