I think I did this correctly. If you want to check, here it is… JAX/NUMPYRO does not include constant_data · Issue #5781 · pymc-devs/pymc (github.com)
Thank you for the sample code.
I think I did this correctly. If you want to check, here it is… JAX/NUMPYRO does not include constant_data · Issue #5781 · pymc-devs/pymc (github.com)
Thank you for the sample code.