You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Right now everything in nutpie (the sampler and the logp functions) are working in float64. For some models we could get decent speedups if we also support float32.
Pytensor can already compile float32 logp functions, but we have to change the python wrapper to make use of that. We also have to add a <F: Float> type parameter to some operations in nuts-rs, and figure out how to do the simd functions with that.
The text was updated successfully, but these errors were encountered:
Right now everything in nutpie (the sampler and the logp functions) are working in float64. For some models we could get decent speedups if we also support float32.
Pytensor can already compile float32 logp functions, but we have to change the python wrapper to make use of that. We also have to add a
<F: Float>
type parameter to some operations innuts-rs
, and figure out how to do the simd functions with that.The text was updated successfully, but these errors were encountered: