>>> torch.cuda.is_bf16_supported()
True