introduce custom_batching.sequential_vmap
#9275
Merged
custom_batching.sequential_vmap
#9275