nutpie icon indicating copy to clipboard operation
nutpie copied to clipboard

Python wrapper for nuts-rs

Results 48 nutpie issues
Sort by recently updated
recently updated
newest added

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...

help wanted
good first issue

If a deterministic variable uses a RandomVariable, it won't properly update the seed in the `expanding_function` of the compiled pymc model, as we only use the numba function but do...

I figured out that you have to prefix the model name followed by two colons when updating variables in the `with_data` keyword arguments. I don't know if this is intended,...

The barebones PyTorch backend should be in in the next PyTensor release and subsequent PyMC. I think the best way to test it out and gather info on what Ops...

I am trying to run a PyMC model with the nutpie sampler, but initialization seems to be very slow and sampling does not start with these Warning statements: ``` C:\Users\TsubasaTakaya\miniconda3\envs\pymc_env\Lib\site-packages\pytensor\link\numba\dispatch\basic.py:379:...

Otherwise it seems this can lead to the stan makefile not recognizing the compiler: ``` RuntimeError: Command make STAN_THREADS=true STANCFLAGS=--include-paths=. /tmp/tmp9av6ksrm/model_model.so failed with code 2. stdout: stderr: stan/lib/stan_math/make/libraries:117: *** "Need...

updates: - [github.com/astral-sh/ruff-pre-commit: v0.5.6 → v0.6.7](https://github.com/astral-sh/ruff-pre-commit/compare/v0.5.6...v0.6.7)

Is there a way to disable the jittering of the initial point on each chain when sampling? I'm working on quite fussy models, and would like finer control over the...

enhancement
question

Hi! I think it is fairly common for a user to have a log density/its gradient _outside_ of a PyMC or Stan model. I was helping one such [user on...

This exports a torch module (or function) using dynamo_export to onnx, and then uses the onnxruntime to sample from the model. Still WIP, not tested much and the interface is...