Metadata-Version: 2.2
Name: walnutpie
Version: 0.0.1
Summary: Walnuts sampling with Nutpie adaptation
Author-Email: Brian Ward <bward@flatironinstitute.org>, Bob Carpenter <bcarpenter@flatironinstitute.org>
License: MIT License
Project-URL: Homepage, https://github.com/flatironinstitute/walnutpie
Project-URL: Bug Tracker, https://github.com/flatironinstitute/walnutpie/issues
Requires-Python: >=3.9
Requires-Dist: numpy
Provides-Extra: stan
Requires-Dist: bridgestan>=2.9.0; extra == "stan"
Requires-Dist: stanio; extra == "stan"
Provides-Extra: compiled
Requires-Dist: numba; extra == "compiled"
Provides-Extra: test
Requires-Dist: pytest; extra == "test"
Requires-Dist: scipy; extra == "test"
Requires-Dist: bridgestan; extra == "test"
Provides-Extra: test-fast
Requires-Dist: pytest; extra == "test-fast"
Requires-Dist: numba; extra == "test-fast"
Requires-Dist: numba_stats; extra == "test-fast"
Requires-Dist: bridgestan; extra == "test-fast"
Description-Content-Type: text/markdown

# walnutpie - the Within-orbit Adaptive Leapfrog No-U-Turn Sampler

## Documentation

Documentation for `walnutpie` can be found on [Github Pages](https://flatironinstitute.github.io/walnutpie/latest/).

## Basic usage

```python
import walnutpie as wp


def logp(x):
    # your code here
    lp = ...
    gradient = ...
    return lp, gradient


draws = wp.walnuts_pyfunc(logp, num_params=10)

print(wp.ess(draws))
```

## Prior work

The design of this library is heavily influenced by both
[TinyStan](https://github.com/WardBrian/tinystan) and [Nutpie](https://github.com/pymc-devs/nutpie).
