`AveragedModel` and FSDP2

Is there a way to combine AveragedModel with modern ND parallel or at least FSDP2 with DTensor?