├── .github └── workflows │ └── build.yaml ├── .gitignore ├── .pre-commit-config.yaml ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── _assets └── _logo.png ├── benchmarks ├── README.md ├── cuda_benchmark_results.npz ├── data │ ├── fn_1e1.casadi │ ├── fn_1e2.casadi │ ├── fn_1e3.casadi │ ├── fn_1e4.casadi │ └── fn_1e5.casadi ├── jaxadi_benchmark_results.npz ├── jaxadi_vs_cusadi.ipynb ├── run_benchmark.py └── visualize.ipynb ├── docs ├── README.md ├── index.html └── static │ ├── css │ ├── bulma-carousel.min.css │ ├── bulma-slider.min.css │ ├── bulma.css.map.txt │ ├── bulma.min.css │ ├── fontawesome.all.min.css │ └── index.css │ ├── images │ ├── compare_1e1_bar.png │ ├── compare_1e2_bar.png │ ├── compare_1e3_bar.png │ ├── compare_1e4_bar.png │ ├── favicon.png │ ├── favicon.svg │ ├── logo.png │ ├── speedup_ratio.png │ └── steve.webm │ └── js │ ├── bulma-carousel.js │ ├── bulma-carousel.min.js │ ├── bulma-slider.js │ ├── bulma-slider.min.js │ ├── fontawesome.all.min.js │ └── index.js ├── environment.yml ├── examples ├── 00_translate.py ├── 01_lower.py ├── 02_convert.py ├── 03_pendulum_rollout.py ├── 04_pinocchio.py ├── 05_mjx.py └── _demo.ipynb ├── jaxadi ├── __init__.py ├── _compile.py ├── _convert.py ├── _declare.py ├── _expand.py ├── _graph.py ├── _ops.py └── _preprocess.py ├── pyproject.toml └── tests ├── test_casadi_equality.py ├── test_examples.py └── test_input.py /.github/workflows/build.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/.github/workflows/build.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/README.md -------------------------------------------------------------------------------- /_assets/_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/_assets/_logo.png -------------------------------------------------------------------------------- /benchmarks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/benchmarks/README.md -------------------------------------------------------------------------------- /benchmarks/cuda_benchmark_results.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/benchmarks/cuda_benchmark_results.npz -------------------------------------------------------------------------------- /benchmarks/data/fn_1e1.casadi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/benchmarks/data/fn_1e1.casadi -------------------------------------------------------------------------------- /benchmarks/data/fn_1e2.casadi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/benchmarks/data/fn_1e2.casadi -------------------------------------------------------------------------------- /benchmarks/data/fn_1e3.casadi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/benchmarks/data/fn_1e3.casadi -------------------------------------------------------------------------------- /benchmarks/data/fn_1e4.casadi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/benchmarks/data/fn_1e4.casadi -------------------------------------------------------------------------------- /benchmarks/data/fn_1e5.casadi: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/benchmarks/data/fn_1e5.casadi -------------------------------------------------------------------------------- /benchmarks/jaxadi_benchmark_results.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/benchmarks/jaxadi_benchmark_results.npz -------------------------------------------------------------------------------- /benchmarks/jaxadi_vs_cusadi.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/benchmarks/jaxadi_vs_cusadi.ipynb -------------------------------------------------------------------------------- /benchmarks/run_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/benchmarks/run_benchmark.py -------------------------------------------------------------------------------- /benchmarks/visualize.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/benchmarks/visualize.ipynb -------------------------------------------------------------------------------- /docs/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/README.md -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/index.html -------------------------------------------------------------------------------- /docs/static/css/bulma-carousel.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/css/bulma-carousel.min.css -------------------------------------------------------------------------------- /docs/static/css/bulma-slider.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/css/bulma-slider.min.css -------------------------------------------------------------------------------- /docs/static/css/bulma.css.map.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/css/bulma.css.map.txt -------------------------------------------------------------------------------- /docs/static/css/bulma.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/css/bulma.min.css -------------------------------------------------------------------------------- /docs/static/css/fontawesome.all.min.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/css/fontawesome.all.min.css -------------------------------------------------------------------------------- /docs/static/css/index.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/css/index.css -------------------------------------------------------------------------------- /docs/static/images/compare_1e1_bar.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/images/compare_1e1_bar.png -------------------------------------------------------------------------------- /docs/static/images/compare_1e2_bar.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/images/compare_1e2_bar.png -------------------------------------------------------------------------------- /docs/static/images/compare_1e3_bar.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/images/compare_1e3_bar.png -------------------------------------------------------------------------------- /docs/static/images/compare_1e4_bar.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/images/compare_1e4_bar.png -------------------------------------------------------------------------------- /docs/static/images/favicon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/images/favicon.png -------------------------------------------------------------------------------- /docs/static/images/favicon.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/images/favicon.svg -------------------------------------------------------------------------------- /docs/static/images/logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/images/logo.png -------------------------------------------------------------------------------- /docs/static/images/speedup_ratio.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/images/speedup_ratio.png -------------------------------------------------------------------------------- /docs/static/images/steve.webm: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/images/steve.webm -------------------------------------------------------------------------------- /docs/static/js/bulma-carousel.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/js/bulma-carousel.js -------------------------------------------------------------------------------- /docs/static/js/bulma-carousel.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/js/bulma-carousel.min.js -------------------------------------------------------------------------------- /docs/static/js/bulma-slider.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/js/bulma-slider.js -------------------------------------------------------------------------------- /docs/static/js/bulma-slider.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/js/bulma-slider.min.js -------------------------------------------------------------------------------- /docs/static/js/fontawesome.all.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/js/fontawesome.all.min.js -------------------------------------------------------------------------------- /docs/static/js/index.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/docs/static/js/index.js -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/environment.yml -------------------------------------------------------------------------------- /examples/00_translate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/examples/00_translate.py -------------------------------------------------------------------------------- /examples/01_lower.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/examples/01_lower.py -------------------------------------------------------------------------------- /examples/02_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/examples/02_convert.py -------------------------------------------------------------------------------- /examples/03_pendulum_rollout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/examples/03_pendulum_rollout.py -------------------------------------------------------------------------------- /examples/04_pinocchio.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/examples/04_pinocchio.py -------------------------------------------------------------------------------- /examples/05_mjx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/examples/05_mjx.py -------------------------------------------------------------------------------- /examples/_demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/examples/_demo.ipynb -------------------------------------------------------------------------------- /jaxadi/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/jaxadi/__init__.py -------------------------------------------------------------------------------- /jaxadi/_compile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/jaxadi/_compile.py -------------------------------------------------------------------------------- /jaxadi/_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/jaxadi/_convert.py -------------------------------------------------------------------------------- /jaxadi/_declare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/jaxadi/_declare.py -------------------------------------------------------------------------------- /jaxadi/_expand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/jaxadi/_expand.py -------------------------------------------------------------------------------- /jaxadi/_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/jaxadi/_graph.py -------------------------------------------------------------------------------- /jaxadi/_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/jaxadi/_ops.py -------------------------------------------------------------------------------- /jaxadi/_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/jaxadi/_preprocess.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/pyproject.toml -------------------------------------------------------------------------------- /tests/test_casadi_equality.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/tests/test_casadi_equality.py -------------------------------------------------------------------------------- /tests/test_examples.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/tests/test_examples.py -------------------------------------------------------------------------------- /tests/test_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/based-robotics/jaxadi/HEAD/tests/test_input.py --------------------------------------------------------------------------------