How do I manipulate shape when using sample_numpyro_nuts

not yet. :joy: