PyTorch renaming submod indices leading to assert break

[kwen2501]

After certain torch 2.2.0.dev version, submod_0, submod_1, submod_2 ... are named as submod_0, submod_2, submod_4 ...

Hence this assert would fail:

pippy/IR.py", line 682, in _number_and_count_forward_stages
    assert all(i in found_idxs for i in range(num_stages, step=2))