GitHub repository: yixiaoer/einshard
Primary language: Python
Description: Einsum-like high-level array sharding API for JAX