Joery
Joery
I'm not sure whether this is possible, but I found out quite awkwardly that `vmapping` solvers strictly requires the optimized function to be staged. For example, this code works fine:...
Possibly related to deepcopy of Gym.Env or GymState objects as this issue is absent for the boardgames.
Update Unit tests or remove from repository.
Hi, I accidentally stumbled upon a problem within the [tutorial notebook](https://colab.research.google.com/github/deepmind/acme/blob/master/examples/tutorial.ipynb) when playing around with the acme and reverb API that causes a weird synchronization behaviour between sampling from the...
### Describe the bug When I log a table or a plot, I can easily recover the logged entry inside `run.scan_history`. When I inspect an individual entry, this gives me...
### Description I have logged **a lot of data** in my W&B projects, and now I want to manipulate my logs to make nice plots. To do this, I first...
For domain randomization it is not particularly easy to `vmap` over different `System` values. For example the `gravity` values, or the `elasticity`. Preferably you should be able to do this...
Hi, when calling `hk.transform_with_state` internally there is an access to `jax.xla`, this is marked as deprecated in favor of `jax.interpreters.xla`. The problem is in checking if the provided function `f`...
Hi, I was wondering if a type-annotation for Batched datastructures exists or if it could be implemented? So something similar along the likes of `PyTree` but for indicating that all...
Hi, I was wondering if there were plans to by default support `tree` like structures in dm_env.specs. This would be extremely relevant to jax-based environments, or generic environments utilizing `deepmind/tree`,...