Fix tree_multimap & greedy device grabbing

RobertTLange released this 24 Aug 10:34

· 107 commits to main since this release

6682836

[v0.0.5] - 24/08/2022

Fixed

Fix deprecated tree_multimap to simple tree_map.
Fix device grabbing when using an jnp.array to set default in Acrobot env.

Assets 2