Closed rhayes777 closed 3 months ago
Need to check JAX and search chaining. We mostly do config on the type -> prior relationship so it was hard to see how to carry this over
I would make it so that the config default gives the same prior to everything on the numpy array.
We can worry about JAX and search chaining later on then!
I would make it so that the config default gives the same prior to everything on the numpy array.
We can worry about JAX and search chaining later on then!
So at the moment you define one prior and that gets copied to every slot. Would we want a config to determine the default for all Arrays globally?
Should now work with prior passing and JAX
Adds the Array class which functions as a PriorModel that creates numpy arrays of floats.
An array is defined by its shape and a prior.
The prior is copied to each index meaning that a 2x2 array has four independent priors.
Arrays can be accessed and modified using indexing.
They can be instantiated just as any other model class