Enzyme-JAX
Enzyme-JAX copied to clipboard
Rotate of Slice / Wrap of Slice
If you have an op with multiple slices which have rotate users
we should make one rotate on a larger slice thereof, and slice into the rotate (essentially we want to perform the rotate on as large a tensor as possible, thereby reducing comms).
same thing for a wrap