Float32 not working with pytensor, worked with aesara

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