[functorch] Added temporary odict pytree registration (pytorch/functorch#544)
* Added temporary odict pytree registration
Description:
- Added temporary odict pytree registration
- combine_state_for_ensemble can work on torchvision segmentation models
Related to https://github.com/pytorch/functorch/issues/445#issuecomment-1050946177
* Added pytree odict test
