https://github.com/NVlabs/storm
Stochastic Tensor Optimization for Robot Motion - A GPU Robot Motion Toolkit
https://github.com/NVlabs/storm
Last synced: 3 months ago
JSON representation
Stochastic Tensor Optimization for Robot Motion - A GPU Robot Motion Toolkit
- Host: GitHub
- URL: https://github.com/NVlabs/storm
- Owner: NVlabs
- License: other
- Created: 2021-04-27T20:54:27.000Z (about 4 years ago)
- Default Branch: main
- Last Pushed: 2022-11-08T22:59:03.000Z (over 2 years ago)
- Last Synced: 2023-10-20T22:08:42.532Z (over 1 year ago)
- Language: Python
- Size: 13.1 MB
- Stars: 139
- Watchers: 7
- Forks: 30
- Open Issues: 5
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
- awesome-isaac-gym - Code
README
# STORM
**Stochastic Tensor Optimization for Robot Motion** - *A GPU Robot Motion Toolkit*[[Install Instructions](install_instructions.md)] [[Paper](https://arxiv.org/abs/2104.13542)] [[Website](https://sites.google.com/view/manipulation-mpc/home)]
This package contains code for reactive robot motion leveraging parallel compute on the GPU. The implemented control framework leverages MPPI to optimize over sampled actions and their costs. The costs are computed by rolling out the forward model from the current state with the sampled actions. Most files are documented with sphinx. Once you clone this repo, go into docs folder and run `sh generate_docs.sh` to generate documentation.
**To run on a real Franka Panda, you can use this low-level control wrapper:** [franka_motion_control](https://github.com/mohakbhardwaj/franka_motion_control) from [Mohak Bhardwaj](https://github.com/mohakbhardwaj).
![]()
![]()
## Updates
Jan. 2022 - Add CoRL citation, merge torch.size() bug (thanks [@maxpahn](https://github.com/maxspahn)).## Contributors
- Mohak Bhardwaj
- Balakumar Sundaralingam## Citation
If you use this source code, please cite the below article,```
@article{storm2021,
title={{STORM}: An Integrated Framework for Fast Joint-Space Model-Predictive Control for Reactive Manipulation},
author={Mohak Bhardwaj and Balakumar Sundaralingam and Arsalan Mousavian and Nathan D. Ratliff and Dieter Fox and Fabio Ramos and Byron Boots},
booktitle={5th Annual Conference on Robot Learning },
year={2021},}
```## Contributing to this code
Refer to CLA before making contributions.