flax
allow custom dot_general for einsum.
#3884
Merged

Commits
  • allow custom dot_general for einsum.
    a-googler committed 1 year ago
Loading