Did you cast all your input data to float32? You can use pm.floatX(data) to auto-cast.
pm.floatX(data)