Add einops-style `rearrange` - Githubissues

stanford-crfm / haliax

Named Tensors for Legible Deep Learning in JAX

Apache License 2.0

149 stars 11 forks source link

Add einops-style `rearrange` #44

Closed dlwh closed 11 months ago