Numerical derivatives of dynamics for mujoco

Forked from https://github.com/wecacuee/mujoco_py_deriv

• Needs mujoco licence to run.
• Wraps derivative.cpp to call from Python.

Installation

1. Install mujoco_py
2. pip install mujoco-py-derivatives

Usage

Prepare mujoco model.

import mujoco_py as mj
from mujoco_py_derivatives import MjDerivative, checkderiv

# Prepare mujoco model and data
model = mj.load_model_from_path("flat_pusher_sample.xml")
sim = mj.MjSim(model, nsubsteps=nstep)
dmain = sim.data

Compute numerical derivative

# To compute δf/δx
f = ["qacc"]
x = ["qfrc_applied", "qvel", "qpos"]
deriv_obj = MjDerivative(model, dmain, f, x)
deriv = deriv_obj.compute()