-
For background see https://github.com/google/jax/discussions/13555
`jax.experimental.host_callback` is deprecated, and we should switch to using `jax.debug.print`/`jax.debug.callback` in [blackjax/…
-
combine Kalman Filter from dynamax and sequential monte carlo from [Blackjax](https://github.com/blackjax-devs/blackjax) to implement RBPF for switching linear Gaussian SSM. For details, see sec 13.4.…
-
It would be good to reimplement these demos, used in https://probml.github.io/pml-book/book2
- https://github.com/probml/JSL/blob/main/jsl/demos/bootstrap_filter.py
- https://github.com/probml/JSL…
-
- [ ] Everything is a kernel
- [ ] The 3-layer design
- [ ] Programmable inference
rlouf updated
3 months ago
-
### Current behavior
Blackjax doesn't have code for constraints.
Stan (and NumPyro) handle constraints by transformation, and one could extract the Numpyro transformations and use those.
…
-
This example is taken from the baseball case study in `pymc-examples`. We fit a beta-binomial model to some baseball batting data:
```python
data = pd.read_csv(pm.get_data("efron-morris-75-data.t…
-
### Describe the issue as clearly as possible:
Using `jax`/`jaxlib` 0.4.28 and `jaxopt` 0.8.3, the following test fails on `aarch64-linux`:
```
FAILED tests/adaptation/test_adaptation.py::test_chee…
-
https://jax-ml.github.io/bayeux/
-
The VIAlgorithm class has code and documentation that do not match in `base.py` [here](https://github.com/blackjax-devs/blackjax/blob/main/blackjax/base.py).
The documentation indicates this class …
-
### Description
Segmentation fault when calling jax.lax.scan(), jax.lax.map(), and related functions. Segmentation fault can be traced back to core.AxisPrimitive().bind() call. Reproducible using jax…