Joery

Issue also observed with MuZero agents. Although memory grows much slower than with AlphaZero.

> @gil2rok @joeryjoery Just to make sure that I understand the feature request correctly. Would you say it is the same as doing something like the following: > > ```...

Forgot to clarify. I'm aware that tuple, dicts, etc. are already supported as valid specs. The reason why I'm suggesting an explicit Tree spec is for my type-checker to OK...

Hey I stumbled upon this, if anyone finds it useful I had to implement Lambert's W for the k=0th branch for real inputs only. ```python from functools import partial import...

Hey thanks for the reply, I still don't quite get it so I'll post the full method, ```python import wandb api = wandb.Api() run = api.run(...) export = run.scan_history(keys=['my_table_key']) #...

To clarify/ simplify, I would like to know how I can get the artifact qualified name (or just the artifact): ```python 'joery/Regression/run-ng2kn18n-EvaluationServicetest_functional_metricsfunctionsFourier0table:v0' ``` from the logged table entry inside the...

I solved the problem by coincidence, I'll post the way that I achieved it so someone else can find this (preferably you guys can add this to the documentation). The...

Another quick fix that I use is to wrap the reverb dataset inside the following class: ```python class RefreshIterator: """tf.data.Dataset fix for slow reverb client synchronization. Wrap around reverb-dataset.""" __slots__...

Wanted to add an example of another workaround: https://github.com/automl/CARL. In this library for meta-RL, instead of batching environments on the GPU which Brax should support, the CARL-brax environments create VectorizedWrappers...

> Hi @joeryjoery , I believe we considered passing around sys as part of the env state, but IIRC we managed to squeeze out better performance using the current implementation....