Fixing batching over sparse arrays

infer-actively / pymdp

A Python implementation of active inference for Markov Decision Processes

MIT License

419 stars 83 forks source link

Fixing batching over sparse arrays #147

Closed dimarkov closed 2 weeks ago

dimarkov commented 2 weeks ago

I have made possible batching over sparse arrays with vmap, and also merged into single function both sparse and dense array computations for backward smoothing.