yixiaoer / einshard

Einsum-like high-level array sharding API for JAX
34Updated 8 months ago

Alternatives and similar repositories for einshard:

Users that are interested in einshard are comparing it to the libraries listed below