You can also use the warn_float64 flag to see where are float64 being introduced: config – PyTensor Configuration — PyTensor dev documentation
You can also use the warn_float64 flag to see where are float64 being introduced: config – PyTensor Configuration — PyTensor dev documentation