move task.py to jax/envs/env.py

infer-actively / pymdp

A Python implementation of active inference for Markov Decision Processes

MIT License

419 stars 83 forks source link

Closed tverbele closed 3 weeks ago

tverbele commented 3 weeks ago

Move the jax/task.py to jax/envs/env.py to mirror the numpy backend.