├── .github └── workflows │ ├── formatting.yml │ └── pyright.yml ├── .gitignore ├── examples ├── 01_kinematics.py ├── 02_collbody.py ├── 03_kin_with_coll.py ├── 04_trajopt.py ├── 05_mpc_gradient.py ├── 06_mpc_sampling.py └── assets │ └── yumi_trajectory.npy ├── pyproject.toml ├── readme.md ├── scripts └── profile_ik.py └── src └── jaxmp ├── __init__.py ├── coll ├── __init__.py ├── _collide.py ├── _collide_types.py └── _robot_coll.py ├── extras ├── __init__.py ├── grasp_antipodal.py ├── solve_ik.py ├── solve_mpc.py └── urdf_loader.py ├── kinematics.py └── robot_factors.py /.github/workflows/formatting.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/.github/workflows/formatting.yml -------------------------------------------------------------------------------- /.github/workflows/pyright.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/.github/workflows/pyright.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/.gitignore -------------------------------------------------------------------------------- /examples/01_kinematics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/examples/01_kinematics.py -------------------------------------------------------------------------------- /examples/02_collbody.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/examples/02_collbody.py -------------------------------------------------------------------------------- /examples/03_kin_with_coll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/examples/03_kin_with_coll.py -------------------------------------------------------------------------------- /examples/04_trajopt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/examples/04_trajopt.py -------------------------------------------------------------------------------- /examples/05_mpc_gradient.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/examples/05_mpc_gradient.py -------------------------------------------------------------------------------- /examples/06_mpc_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/examples/06_mpc_sampling.py -------------------------------------------------------------------------------- /examples/assets/yumi_trajectory.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/examples/assets/yumi_trajectory.npy -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/pyproject.toml -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/readme.md -------------------------------------------------------------------------------- /scripts/profile_ik.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/scripts/profile_ik.py -------------------------------------------------------------------------------- /src/jaxmp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/src/jaxmp/__init__.py -------------------------------------------------------------------------------- /src/jaxmp/coll/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/src/jaxmp/coll/__init__.py -------------------------------------------------------------------------------- /src/jaxmp/coll/_collide.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/src/jaxmp/coll/_collide.py -------------------------------------------------------------------------------- /src/jaxmp/coll/_collide_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/src/jaxmp/coll/_collide_types.py -------------------------------------------------------------------------------- /src/jaxmp/coll/_robot_coll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/src/jaxmp/coll/_robot_coll.py -------------------------------------------------------------------------------- /src/jaxmp/extras/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/src/jaxmp/extras/__init__.py -------------------------------------------------------------------------------- /src/jaxmp/extras/grasp_antipodal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/src/jaxmp/extras/grasp_antipodal.py -------------------------------------------------------------------------------- /src/jaxmp/extras/solve_ik.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/src/jaxmp/extras/solve_ik.py -------------------------------------------------------------------------------- /src/jaxmp/extras/solve_mpc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/src/jaxmp/extras/solve_mpc.py -------------------------------------------------------------------------------- /src/jaxmp/extras/urdf_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/src/jaxmp/extras/urdf_loader.py -------------------------------------------------------------------------------- /src/jaxmp/kinematics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/src/jaxmp/kinematics.py -------------------------------------------------------------------------------- /src/jaxmp/robot_factors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chungmin99/jaxmp/HEAD/src/jaxmp/robot_factors.py --------------------------------------------------------------------------------