Support batching in MLIR autodiff operations (2nd try) #6606

Job	Run time
Enzyme CI LLVM 15 Release ubuntu-22.04	4m 39s
Enzyme CI LLVM 15 Release macOS	3m 20s
Enzyme CI LLVM 15 Debug ubuntu-22.04	4m 5s
Enzyme CI LLVM 15 Debug macOS	2m 46s
Enzyme CI LLVM 16 Release ubuntu-22.04	4m 28s
Enzyme CI LLVM 16 Release macOS	4m 53s
Enzyme CI LLVM 16 Debug ubuntu-22.04	3m 43s
Enzyme CI LLVM 16 Release macOS XCode	2m 42s
Enzyme CI LLVM 16 Debug macOS	2m 53s
	33m 29s

Provide feedback