einops

1 个项目 · ⭐ 9.6k

arogozhnikov/einops

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

Python ⭐ 9.6k ⑂ 397 MIT · 2026-07-05推送

deep-learning jax

2026-07-05