Compatibility Patches

PyTorch version compatibility and fastai backport patches for fastxtend