Take it down to float32 to reduce memory usage