search

infer-actively / pymdp

A Python implementation of active inference for Markov Decision Processes
MIT License
450 stars 89 forks source link

Agent jax vmap agent class #103

Closed conorheins closed 1 year ago

conorheins commented 1 year ago

@dimarkov I vote we merge this branch back into the "main" agent_jax branch