mjx

Here are 4 public repositories matching this topic...

A simple JAX-based implementation of random search for locomotion tasks using MuJoCo XLA (MJX).

learning robotics optimization parallel-computing locomotion random-search mujoco jax motor-control mjx

Highly customizable (and fast) recurrent RL library for JAX/MJX

reinforcement-learning robotics recurrent-neural-networks mujoco ppo a2c jax mjx

Code for the paper "Decentralized neural dynamics and sensory constraints shape brittle star locomotion"

Add a description, image, and links to the mjx topic page so that developers can more easily learn about it.

To associate your repository with the mjx topic, visit your repo's landing page and select "manage topics."