search

infer-actively / pymdp

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

Agent jax #70

Closed dimarkov closed 2 years ago

dimarkov commented 2 years ago

Initial changes to the Agent class for integration with jax backend