bandits

bandits copied to clipboard

replace jax.ops.index_update with jax.numpy.at

1

jax.ops.index_update has been removed (see https://github.com/google/jax/commit/f51a05a889f2fcb19946352b9d65f2b6c49fec4a) which breaks all the bandit code. Please use [jnp.at()](https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.ndarray.at.html) instead.

murphyk

Make them stateful

gerdm

- [ ] Move `results/*` out of main bandits' folder

gerdm

- [ ] Move pre-defined architectures into an `architectures/` folder - [ ] Move demos outside main bandits project. Create new `demos/` folder - [ ] Tests should go into...

gerdm

Refactor JSL dependency to dynamax and rebayes.

gerdm