einopx: A unified JAX-native array pattern API

A_K_Nain · x · 2026-09-02

einopx is a unified JAX-native array pattern API that integrates shape checks, sharding checks, reshape, rearrange, reshard, and einsum contractions into a single operation.

Original post →

More from Infra

Infra channel →