Vectorization / scalarization exercise #26
Description
Grabbing the custom Sum Op from first session, add vectorization dispatch and subtensor rewrite for "scalarization"
{{ message }}
Grabbing the custom Sum Op from first session, add vectorization dispatch and subtensor rewrite for "scalarization"