brax

brax copied to clipboard

Add support for initial alpha value for SAC

1

This PR adds the option to specify the $\alpha$, the initial temperature value for SAC. Additionally, I have added a detailed docstring for the SAC `train` function.

varunagrawal

Insecure Deserialization attack on pickle.loads

2

I noticed that there is a module named `load_params` that passes user input directly into pickle , which allows an attacker to execute system commands with insecure deserialization attack on...

omidxrz

Tendon/Muscle Rendering in Brax Visualizer

4

Hello brax team, I am exploring the integration of musculoskeletal model simulations using [myosuite](https://github.com/myohub/myosuite) with brax, and I would like to see tendon and muscle features within the brax simulation...

ttktjmt

`_generate_eval_unroll` throws "INTERNAL: the requested functionality is not supported"

9

I am trying to run a simple SAC example for quadruped locomotion. The script I use is ```python env_name = 'barkour' env = envs.get_environment(env_name) num_timesteps = 100000 train_fn = functools.partial(...

varunagrawal

Support for Flax NNX API

1

Hi Brax development team, We are currently using the `brax.training.ppo` training scripts in our project and have been very happy with Brax's performance. We noticed that the `brax.training` module uses...

scott-yj-yang

See https://github.com/google/brax/discussions/536

btaba