`torch_dtype` is deprecated! Use `dtype` instead!