[jagged] Get more meaningful inputs for jagged reduction kernels

[xuzhao9]

The current shapes are very much contrived, obtain more meaningful shapes for jagged

[Jokeren]

@Darkviper7

We are looking at exactly the same issue. Let me know if you have any insights and probably you can offload the task to the student.

[Jokeren]

@Darkviper7

Please update the jagged operators you plan to explore. Xu is going to assist us on this issue.

[Darkviper7]

I plan to explore jagged_sum, jagged_mean, jagged_softmax operators.

[xuzhao9]

Update: we have finished collecting production jagged reduction shapes at 352be40. Now we are going through the process of requesting to open source them.

cc @Darkviper7 @Jokeren

[Jokeren]

Thank you! That would be very helpful. @Darkviper7 please let us know if there are other ops that you're interested in.

[Darkviper7]

@xuzhao9 Hello, Thank you. I was wondering when the production shapes will be available and if there is any update.

[xuzhao9]

Hi @Darkviper7, we have extracted production shapes in 352be40, but right now it is internal only. I am still working out how to open-source it.