Float32 not working with pytensor, worked with aesara

4 
import pandas as pd
import pymc as pm
import pymc.sampling_jax
import pandas as pd
import random
random.seed(1)

import os
os.environ["XLA_PYTHON_CLIENT_ALLOCATOR"] = "platform"
import pytensor
pytensor.config.floatX="float32"
pytensor.config.warn_float64="raise"
import jax
jax.default_backend()
df = pd.read_csv("minimal.csv")

my_model = pm.Model()
with my_model:
    Intercept = pm.Normal("Intercept", mu=0, sigma=5)
    theta_as_logit = Intercept
    y = pm.ConstantData("y", df["Y"] == True)
    theta_as_probability = pm.invlogit(theta_as_logit)
    Y_obs = pm.Bernoulli("Y_obs", p=theta_as_probability, observed=y)
    my_fit_pymc = pm.sampling_jax.sample_numpyro_nuts(random_seed=1234, tune=1000,
         draws=20, target_accept=0.47, chains = 2, chain_method='parallel', 
         idata_kwargs=dict(log_likelihood=False), 
         postprocessing_backend="cpu")

minimal.csv attached
minimal.csv (129 Bytes)