VI is sensitive to float32 and might break. Currently it it is [missing](https://github.com/pymc-devs/pymc/blob/741d45540016ab52a15ad5bff521614b6ae1d835/.github/workflows/tests.yml#L350) from the float32 test suit