Also, for the other parameter use float32 to use different dtype.

Also, with int32, make sure to draw the random value correctly. We where drawing only 0s.