kinax A lightweight implementation of Differentiable FK from URDF in Jax. This is implemented on top of Brax. Installation Simply do conda activate <your_env> && pip install -e . Example Run python examples/kinematics.py