brax issues

Persistent Caching of Jitted Functions on GPU for Brax Envs and Autodiff

Hi, I am trying to use persistent caching in XLA on GPU to speed up the execution of my Brax code. Tracking [issues](https://github.com/google/jax/discussions/13736#discussioncomment-5903999) in JAX this seems to be possible...

bebark

Radians vs degrees in quaternion logic

1

Hi Brax, When converting between quaternions and Euler angles, the function [euler_to_quat](https://github.com/google/brax/blob/673a41f780fe0f137507d1c35286577c48bde4d1/brax/math.py#L69) takes the Euler angles in degrees, but the inverse function, [quat_to_euler](https://github.com/google/brax/blob/673a41f780fe0f137507d1c35286577c48bde4d1/brax/math.py#L81), returns the Euler angles in radians. It...

andrewwarrington

enhancement

good first issue

Rendering after every few epochs at train time

3

Hi! Thanks for opensourcing this amazing piece of software! Currently, I'm trying to learn RL for Robotics and I want to be able to see the physical rendering of an...

aggritvik

Agents: Short-Horizon Actor Critic

16

This is an implementation of https://arxiv.org/pdf/2204.07137.pdf Not sure if there is interest merging this into the main branch. This might be an algorithm worth supporting as it leverages the differentiable...

peabody124

io/image: Replace PyTinyrenderer with a Pure JAX renderer

7

This solves #331 #108 #47 #67 and is involved in #302 [Example Colab](https://colab.research.google.com/drive/14uV9ORm44lDNxmyjOvpOgGX-o-EVgkhS) (adopt majority of the code from Brax teams's [Brax Training](https://colab.research.google.com/github/google/brax/blob/main/notebooks/training.ipynb)) Known issue: the plane may not display...

JoeyTeng

HTML visualizer follow target and freeze angle features not working in Brax v2

4

imoneoi

Remove duplicate code in brax.envs by using get_environment

1

This is a small pull request simply removing code duplication by reusing the `brax.envs.get_environment` function. I have also made some slight improvements to the documentation of the `brax.envs.create` function to...

arxaqapi

How to get observation space, or at least low/high bounds?

2

Hi, I'm used to having observation_space in gym.Env / VectorEnv classes that tell me how to normalize observations, among other uses. Am I right to think this is missing in...

ikamensh