High-performance inverse kinematics solver optimized for cross-embodiment VLA/AI applications
Project description
EmbodiK
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
ROBOTIS AI Worker constraint teleop
RB-Y1 bimanual whole-body IK
Unitree G1 retargeting IK
Spot full-body IK
Spot locomanipulation mjviser
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
- Installation - platform setup, source builds, and troubleshooting.
- Quickstart - first IK calls and solver concepts.
- Working with Transforms - transform helpers and SE(3) operations.
- Examples - public scripts and development-only demos.
- API Reference - Python API generated from docstrings.
- GPU Solvers - FI-PeSNS and PPH-SNS batch solver notes.
- Development - local build, tests, and contributor workflow.
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
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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distributions
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
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
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
1f71d066a0bdb44f6b211ca63986c8ce965c3c7764048885fb2d9eca9997470b
|
|
| MD5 |
b7d4b23aef9e825cf3824195e0066a99
|
|
| BLAKE2b-256 |
5a83bba068587ff8271d0d3e616ed2c5ca2d85e651354ed023bf6964bcd872e8
|
File details
Details for the file embodik-0.20.9-cp312-cp312-manylinux_2_28_x86_64.whl.
File metadata
- Download URL: embodik-0.20.9-cp312-cp312-manylinux_2_28_x86_64.whl
- Upload date:
- Size: 27.5 MB
- Tags: CPython 3.12, manylinux: glibc 2.28+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.10.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
71a579559caef55c3e16566b6f8a7dff5d9f9533a2cf6326c05367167613415c
|
|
| MD5 |
0d0f8985d2a7731b4e1bb74e346a7050
|
|
| BLAKE2b-256 |
7ca2d87899234aae903fb53050c436addac171d2afbd3d85f275f0e9e9683e37
|
File details
Details for the file embodik-0.20.9-cp312-cp312-macosx_14_0_arm64.whl.
File metadata
- Download URL: embodik-0.20.9-cp312-cp312-macosx_14_0_arm64.whl
- Upload date:
- Size: 23.8 MB
- Tags: CPython 3.12, macOS 14.0+ ARM64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.10.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
8663b3f6ff46724c1aa42b1a827cad80d756db21788a2bc60de54099da3025e6
|
|
| MD5 |
46455473c339bb10d868fa455216d484
|
|
| BLAKE2b-256 |
2e5a65851fa42af2f510774ace2b323bec0723c3fea8df10f65eaa1c982fdd4f
|
File details
Details for the file embodik-0.20.9-cp311-cp311-manylinux_2_28_x86_64.whl.
File metadata
- Download URL: embodik-0.20.9-cp311-cp311-manylinux_2_28_x86_64.whl
- Upload date:
- Size: 27.5 MB
- Tags: CPython 3.11, manylinux: glibc 2.28+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.10.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
03f956087f3ac37e01ded39cbd4299d0e28593d59964031e9f4dba158cb39bce
|
|
| MD5 |
d4778d9f9903106f525f9723e36f9f09
|
|
| BLAKE2b-256 |
9745ebefdfbe42e7973fd97349e1bba72c6e89db8c8878ed078a2baebffddced
|
File details
Details for the file embodik-0.20.9-cp311-cp311-macosx_14_0_arm64.whl.
File metadata
- Download URL: embodik-0.20.9-cp311-cp311-macosx_14_0_arm64.whl
- Upload date:
- Size: 23.8 MB
- Tags: CPython 3.11, macOS 14.0+ ARM64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.10.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
220f6e2e696e8aecd33289e08c0705a5fc0f026784b44bb33832c3248703b5c6
|
|
| MD5 |
e648693630fd636ae4d70815285de91e
|
|
| BLAKE2b-256 |
2a7eb0c99052261032f7c3cbf34d874fd84f1ff32dc12f30a8363bfa7a8ae60e
|
File details
Details for the file embodik-0.20.9-cp310-cp310-manylinux_2_28_x86_64.whl.
File metadata
- Download URL: embodik-0.20.9-cp310-cp310-manylinux_2_28_x86_64.whl
- Upload date:
- Size: 27.5 MB
- Tags: CPython 3.10, manylinux: glibc 2.28+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.10.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
739e2e02ad28ac598f75acb0ab4705d1f8f1432263f7dbea85b705b74f26289b
|
|
| MD5 |
57dc95fb84117081d5ccccf4c75ba422
|
|
| BLAKE2b-256 |
048860c949a75e242f44072f89d82895ceb7b10d2694ef5d7e11ad2d2cacf1e6
|
File details
Details for the file embodik-0.20.9-cp310-cp310-macosx_14_0_arm64.whl.
File metadata
- Download URL: embodik-0.20.9-cp310-cp310-macosx_14_0_arm64.whl
- Upload date:
- Size: 23.8 MB
- Tags: CPython 3.10, macOS 14.0+ ARM64
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.10.12
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
9b517be8d724f827a92ed31d75b3d7611986e5209aef009f5de5769a6d410d3c
|
|
| MD5 |
407db2ae31d779ff2177c0dc2fc41494
|
|
| BLAKE2b-256 |
e557889e33ff28f2c346cfb3d38eb77a06009e0c7e66c8cde2d36f7e8a9018e1
|