The first step should be to retry your setup without DeepSpeed, and if the problem persists, then you can report the issue.