Bas van der Heijden
Bas van der Heijden
Google Colab hangs on installing pybullet with pip with the following output: Running the following in google colab: ``` !pip3 install pybullet --upgrade ``` Results in the following output: ```...
### Description Hi, I am encountering an issue with the `in_axes` parameter when using `jax.vmap`. Specifically, I am passing `in_axes` as a pytree that mirrors the structure of the arguments...
Hello, I need assistance with configuring a scene where a box can rotate (yaw) around its z-axis and also move within the xyz plane. I've included a partial XML definition...
Hi, I am looking for an algorithm that can find the largest common monomorphic subgraph between two graphs. Specifically, I have two graphs G1 and G2, and I would like...
Hi, The current PPO implementation does not seem to account for time limits. While the `EpisodeWrapper` from brax is used, which tracks a truncation flag ([source](https://github.com/google/brax/blob/2329ae76759e37b0b1f1861cf34e5a67d0f7efa8/brax/envs/wrappers/training.py#L100)) in the info dictionary...
Hi, The `mean` and `var` in the `NormalizeVecObservation` wrapper located [here](https://github.com/luchris429/purejaxrl/blob/4384480641f1eb39ee892b6f71ab8fd6c52c1c38/purejaxrl/wrappers.py#L207) are shaped as `(NUM_ENVS, ) + obs.shape`. I think they should be shaped like a single observation, that is,...
Hi, Do you think it would be possible to implement the magic method `__getitem__` for e.g. the circular buffer such that you can neatly index and slice as you would...
Since jax 0.4.12 (could be for other versions>0.4.1) the `is_jitted` function does not properly detect if inside a jit call. ```python import jumpy def f(a): if jumpy.core.is_jitted(): print("JITTED") else: print("NOT...
This PR allows users to take a `view` on `onp.ndarray` that allows users to use the JAX API for array indexing. This means that the `.at[idx].set(val)` API can be used...
Hi, I was wondering whether it was possible to force the use of numpy even though jax was installed. Namely, I would like to run some parts of my code...