When I exceute the below line in Google Colab, it runs without issues:
adaptive = robust_loss_pytorch.adaptive.AdaptiveLossFunction(num_dims = 4, float_dtype=torch.float32, device=“cuda:0”)
Notice the change in the float_dtype parameter. Can you run with the above and see if it helps?