I changed the model so that the knot parameters are sampled from U(0,1) and N(0,1) and then scaled by the length of x, and it works now. I updated the notebook above in case anyone is interested. Thanks for the help @fritzo, @martinjankowiak!
1 Like