feat: greedy loop to batch via loop fission

feat: support non-constant extra indices

fix: only emit new ops if sure

feat: propagate bounds and eliminate noops

chore: run fmt

feat: check for dynamic_slices

feat: unary elementwise working

chore: comments

feat: clear out unwanted no-ops from loop body

feat: elementwise ops are completely supported

chore: add a note

fix: bad rebase

feat: generalize index handling

test: elementwise loop fission

refactor: support dynamicsliceop

feat: sliceinfo generalize

feat: lift ops by batch op

feat: run batching pass

feat: generalize handling for non-reshape case

Author: avik-pal

src/enzyme_ad/jax/Implementations/WhileLoopInfo.h

@@ -39,6 +39,8 @@ struct WhileLoopInfo {
   std::optional<int64_t> getConstantStart();
   std::optional<int64_t> getConstantLimit();
 
+  Value getInductionVariable() { return op.getBody().front().getArgument(0); }
+
   int64_t getConstantNumIters();
   Value getNumIters(OpBuilder &builder);