einx icon indicating copy to clipboard operation
einx copied to clipboard

verified publisher icon fferflo
→

Metadata

Universal Tensor Operations in Einstein-Inspired Notation for Python.

Results 11 einx issues
Sort by recently updated
recently updated
newest added

Segment_sum

1 comment

Is there a way to do the equivalent of segment_sum (and product, softmax, etc) in einx? https://docs.jax.dev/en/latest/_autosummary/jax.ops.segment_sum.html jax.ops.segment_sum — JAX documentation

swamidass avatar swamidass

← Metadata

296
Stars
8
Forks
Watchers

Owner

verified publisher icon fferflo

Metadata

Universal Tensor Operations in Einstein-Inspired Notation for Python.

Back