Skip to main content

High-performance inverse kinematics solver optimized for cross-embodiment VLA/AI applications

Project description

EmbodiK

Python C++ Nanobind Build PyPI Docs License: Apache-2.0 GitHub stars

EmbodiK is a high-performance inverse kinematics library for cross-embodiment robotics and VLA/AI applications. It pairs a C++ core with Python bindings, exposes robot-model utilities without requiring the Python pin package at runtime, and includes interactive examples for collision-aware IK, CoM constraints, teleop, GPU batch solving, and dual-arm coordination.

Overview

EmbodiK is designed for bringing up IK behavior across different robot bodies without rewriting the solver stack for each model. The public examples focus on a practical path:

  • start with the smallest fixed-base IK loop.
  • add collision-aware behavior and visualization.
  • connect the same stepping IK pattern to teleop input.
  • use richer clone-only examples for development and stress testing.

The detailed installation notes, API reference, and example walkthroughs live in the official documentation:

https://robodreamer.github.io/embodik/

Quick Start

Fastest path for most users is the wheel-only PyPI install inside a virtual environment:

python -m venv .venv
source .venv/bin/activate
python -m pip install -U pip
python -m pip install --only-binary=:all: embodik
python -c "import embodik; print(embodik.__version__)"

If that import works, the core package is installed.

If pip cannot find a compatible wheel, use the one-shot source installer for your platform from the Installation Guide. It creates the venv, installs native dependencies, builds EmbodiK, and runs an import smoke test.

To run copied examples from the same venv:

python -m pip install "embodik[examples]"
embodik-examples --copy
cd embodik_examples
python 01_basic_ik_simple.py

Published repaired wheels do not require the Python pin package at runtime. Source builds use pin or a system Pinocchio install as the native library provider; keep that provider installed in the environment used to import EmbodiK.

Examples

The pip-facing examples are intentionally split by purpose:

Script Purpose
01_basic_ik_simple.py Minimal fixed-base IK bring-up for a robot preset or new URDF.
02_collision_aware_IK.py Collision-aware IK behavior demo and advanced tuning surface.
03_teleop_ik.py Small adapter showing how teleop input drives the same IK step.
08_com_constraint_example.py CoM support-polygon constraint visualization.
09_dual_arm_ects.py Dual-arm ECTS and orthogonal coordination modes.
12_bimanual_whole_body_ik.py Bimanual whole-body teleop, defaulting to AI Worker and optionally supporting RB-Y1, with CoM and collision handling.
13_unitree_g1_retargeting_ik.py Unitree G1 whole-body retargeting IK with CoM and optional collision handling.
14_spot_full_body_ik_viser.py Spot full-body IK in regular Viser with arm+torso, torso-only, full-body, and two-stage modes.
15_spot_locomanip_mjviser.py Spot locomanipulation ONNX policy rollout in MuJoCo through mjviser.

Run them from a copied example directory:

python 02_collision_aware_IK.py
python 03_teleop_ik.py
python 12_bimanual_whole_body_ik.py
python 13_unitree_g1_retargeting_ik.py
python 14_spot_full_body_ik_viser.py

The regular Viser Spot full-body IK example uses the standard example dependencies and includes a bundled public Spot-with-arm URDF. The MuJoCo/mjviser locomanipulation example needs the optional mjviser stack and includes a bundled public MuJoCo Menagerie Spot-with-arm MJCF scene. mjviser is the MuJoCo-backed web viewer environment for policy rollout and interactive simulation. mjviser-teleop is the same viewer stack plus the optional Seer/xvisio controller dependencies, so use it only when running --enable-teleop:

python -m pip install "embodik[mjviser]"
embodik-examples --copy
cd embodik_examples
python 15_spot_locomanip_mjviser.py --policy locomanip
python 15_spot_locomanip_mjviser.py --policy locomanip-stationary
# add the optional Seer controller extra when needed:
python -m pip install "embodik[mjviser,teleop]"
python 15_spot_locomanip_mjviser.py --enable-teleop --policy locomanip

From a repository checkout, run the same example from the repository root with the matching Pixi environment:

pixi run -e mjviser spot-locomanip-mjviser --policy locomanip
pixi run -e mjviser-teleop spot-locomanip-mjviser --enable-teleop --policy locomanip

Use a single Pixi environment per run: mjviser for browser GUI control and mjviser-teleop for browser GUI plus Seer controller input. Do not add mjviser after python; it is a Pixi environment name, not a Python module argument. Passing --enable-teleop connects the optional controller; the in-app Enable teleop box starts enabled when the controller connects. The locomanipulation app also has solver preference sliders. Base assist controls how much the solver uses x/y/yaw locomotion while tracking the gripper target: lower values keep more motion in the arm, and higher values let the base help earlier. Arm recovery bias increases the arm posture return while condition-number protection is active, which helps avoid fully stretched arm configurations during loco-manipulation teleop. The default values are tuned for teleop: the base helps on reachable x/y/yaw nudges, and arm recovery stays active without making the high-condition-number solve overly stiff. The mjviser loop rate-limits ONNX policy inference and background IK requests to 50 Hz by default while MuJoCo simulation and rendering continue stepping at the model/viewer rate. The IK worker keeps only the newest pending request so collision-constrained solves cannot build a backlog and starve the sim thread. When collision avoidance is enabled, the default checks the 3 closest active collision pairs with balanced speed/accuracy tuning. Clone-based Spot IK changes can be headlessly checked with:

pixi run -e mjviser-teleop python scripts/spot_locomanip_ik_hardening.py

Most examples default to the Panda preset. Use --robot <key> when a script supports alternate robot presets. See the Examples Guide for the full catalog, helper conventions, and clone-only development examples.

Preview

Franka Panda collision-free IK

Franka Panda collision-free IK preview

ROBOTIS AI Worker constraint teleop

Bimanual whole-body IK preview

RB-Y1 bimanual whole-body IK

RB-Y1 bimanual whole-body IK preview

Unitree G1 retargeting IK

Unitree G1 retargeting IK preview

Spot full-body IK

Spot full-body IK preview

Spot locomanipulation mjviser

Spot locomanipulation mjviser preview

Core Capabilities

  • C++ IK core with Nanobind Python bindings.
  • Hierarchical velocity IK tasks for frames, posture, CoM, and dual-arm coordination.
  • Joint-limit, self-collision, and CoM support-polygon constraints.
  • Solver diagnostics for timing, task scaling, and Jacobian condition-number logging.
  • Lie-group-aware configuration operations for floating-base, quaternion, and continuous joints.
  • Native Pinocchio-backed robot model utilities exposed through EmbodiK bindings.
  • Optional Viser visualization for interactive IK demos.
  • Experimental GPU batch IK and collision tooling for high-throughput research workflows.

Documentation

Development

Use Pixi from a repository clone:

pixi run build
pixi run test
pixi run docs-build

Run examples from the clone:

pixi run python examples/01_basic_ik_simple.py
pixi run python examples/02_collision_aware_IK.py
pixi run python examples/03_teleop_ik.py

Clone-only advanced surfaces live under dev_examples/ and are not copied by embodik-examples --copy.

Repository Layout

embodik/
|-- README.md
|-- cpp_core/
|   |-- include/embodik/
|   `-- src/
|-- python_bindings/
|   `-- src/
|-- python/embodik/
|-- examples/
|-- dev_examples/
|-- docs/
|-- scripts/
`-- test/

Star History

Star History Chart

License

EmbodiK is released under the Apache License 2.0. See LICENSE for details. Binary wheels may bundle permissively licensed native dependencies; see THIRD_PARTY_NOTICES.md.

Developer: Andy Park andypark.purdue@gmail.com

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

embodik-0.20.9.tar.gz (16.7 MB view details)

Uploaded Source

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

embodik-0.20.9-cp312-cp312-manylinux_2_28_x86_64.whl (27.5 MB view details)

Uploaded CPython 3.12manylinux: glibc 2.28+ x86-64

embodik-0.20.9-cp312-cp312-macosx_14_0_arm64.whl (23.8 MB view details)

Uploaded CPython 3.12macOS 14.0+ ARM64

embodik-0.20.9-cp311-cp311-manylinux_2_28_x86_64.whl (27.5 MB view details)

Uploaded CPython 3.11manylinux: glibc 2.28+ x86-64

embodik-0.20.9-cp311-cp311-macosx_14_0_arm64.whl (23.8 MB view details)

Uploaded CPython 3.11macOS 14.0+ ARM64

embodik-0.20.9-cp310-cp310-manylinux_2_28_x86_64.whl (27.5 MB view details)

Uploaded CPython 3.10manylinux: glibc 2.28+ x86-64

embodik-0.20.9-cp310-cp310-macosx_14_0_arm64.whl (23.8 MB view details)

Uploaded CPython 3.10macOS 14.0+ ARM64

File details

Details for the file embodik-0.20.9.tar.gz.

File metadata

  • Download URL: embodik-0.20.9.tar.gz
  • Upload date:
  • Size: 16.7 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.13

File hashes

Hashes for embodik-0.20.9.tar.gz
Algorithm Hash digest
SHA256 1f71d066a0bdb44f6b211ca63986c8ce965c3c7764048885fb2d9eca9997470b
MD5 b7d4b23aef9e825cf3824195e0066a99
BLAKE2b-256 5a83bba068587ff8271d0d3e616ed2c5ca2d85e651354ed023bf6964bcd872e8

See more details on using hashes here.

File details

Details for the file embodik-0.20.9-cp312-cp312-manylinux_2_28_x86_64.whl.

File metadata

File hashes

Hashes for embodik-0.20.9-cp312-cp312-manylinux_2_28_x86_64.whl
Algorithm Hash digest
SHA256 71a579559caef55c3e16566b6f8a7dff5d9f9533a2cf6326c05367167613415c
MD5 0d0f8985d2a7731b4e1bb74e346a7050
BLAKE2b-256 7ca2d87899234aae903fb53050c436addac171d2afbd3d85f275f0e9e9683e37

See more details on using hashes here.

File details

Details for the file embodik-0.20.9-cp312-cp312-macosx_14_0_arm64.whl.

File metadata

File hashes

Hashes for embodik-0.20.9-cp312-cp312-macosx_14_0_arm64.whl
Algorithm Hash digest
SHA256 8663b3f6ff46724c1aa42b1a827cad80d756db21788a2bc60de54099da3025e6
MD5 46455473c339bb10d868fa455216d484
BLAKE2b-256 2e5a65851fa42af2f510774ace2b323bec0723c3fea8df10f65eaa1c982fdd4f

See more details on using hashes here.

File details

Details for the file embodik-0.20.9-cp311-cp311-manylinux_2_28_x86_64.whl.

File metadata

File hashes

Hashes for embodik-0.20.9-cp311-cp311-manylinux_2_28_x86_64.whl
Algorithm Hash digest
SHA256 03f956087f3ac37e01ded39cbd4299d0e28593d59964031e9f4dba158cb39bce
MD5 d4778d9f9903106f525f9723e36f9f09
BLAKE2b-256 9745ebefdfbe42e7973fd97349e1bba72c6e89db8c8878ed078a2baebffddced

See more details on using hashes here.

File details

Details for the file embodik-0.20.9-cp311-cp311-macosx_14_0_arm64.whl.

File metadata

File hashes

Hashes for embodik-0.20.9-cp311-cp311-macosx_14_0_arm64.whl
Algorithm Hash digest
SHA256 220f6e2e696e8aecd33289e08c0705a5fc0f026784b44bb33832c3248703b5c6
MD5 e648693630fd636ae4d70815285de91e
BLAKE2b-256 2a7eb0c99052261032f7c3cbf34d874fd84f1ff32dc12f30a8363bfa7a8ae60e

See more details on using hashes here.

File details

Details for the file embodik-0.20.9-cp310-cp310-manylinux_2_28_x86_64.whl.

File metadata

File hashes

Hashes for embodik-0.20.9-cp310-cp310-manylinux_2_28_x86_64.whl
Algorithm Hash digest
SHA256 739e2e02ad28ac598f75acb0ab4705d1f8f1432263f7dbea85b705b74f26289b
MD5 57dc95fb84117081d5ccccf4c75ba422
BLAKE2b-256 048860c949a75e242f44072f89d82895ceb7b10d2694ef5d7e11ad2d2cacf1e6

See more details on using hashes here.

File details

Details for the file embodik-0.20.9-cp310-cp310-macosx_14_0_arm64.whl.

File metadata

File hashes

Hashes for embodik-0.20.9-cp310-cp310-macosx_14_0_arm64.whl
Algorithm Hash digest
SHA256 9b517be8d724f827a92ed31d75b3d7611986e5209aef009f5de5769a6d410d3c
MD5 407db2ae31d779ff2177c0dc2fc41494
BLAKE2b-256 e557889e33ff28f2c346cfb3d38eb77a06009e0c7e66c8cde2d36f7e8a9018e1

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page