├── .github └── workflows │ ├── build_publish_stannum.yaml │ └── run_tests.yaml ├── .gitignore ├── LICENSE ├── README.md ├── benchmark └── performance_benchmark.py ├── docs ├── .nojekyll ├── 404.html ├── CNAME ├── FontAwesome │ ├── css │ │ └── font-awesome.css │ └── fonts │ │ ├── FontAwesome.ttf │ │ ├── fontawesome-webfont.eot │ │ ├── fontawesome-webfont.svg │ │ ├── fontawesome-webfont.ttf │ │ ├── fontawesome-webfont.woff │ │ └── fontawesome-webfont.woff2 ├── ayu-highlight.css ├── book.js ├── clipboard.min.js ├── complex_numbers.html ├── contribution.html ├── css │ ├── chrome.css │ ├── general.css │ ├── print.css │ └── variables.css ├── elasticlunr.min.js ├── favicon.png ├── favicon.svg ├── fonts │ ├── OPEN-SANS-LICENSE.txt │ ├── SOURCE-CODE-PRO-LICENSE.txt │ ├── fonts.css │ ├── open-sans-v17-all-charsets-300.woff2 │ ├── open-sans-v17-all-charsets-300italic.woff2 │ ├── open-sans-v17-all-charsets-600.woff2 │ ├── open-sans-v17-all-charsets-600italic.woff2 │ ├── open-sans-v17-all-charsets-700.woff2 │ ├── open-sans-v17-all-charsets-700italic.woff2 │ ├── open-sans-v17-all-charsets-800.woff2 │ ├── open-sans-v17-all-charsets-800italic.woff2 │ ├── open-sans-v17-all-charsets-italic.woff2 │ ├── open-sans-v17-all-charsets-regular.woff2 │ └── source-code-pro-v11-all-charsets-500.woff2 ├── highlight.css ├── highlight.js ├── index.html ├── introduction.html ├── mark.min.js ├── print.html ├── searcher.js ├── searchindex.js ├── searchindex.json ├── tin.html ├── tomorrow-night.css └── tube │ ├── advanced_field_construction.html │ ├── basics.html │ └── dynamic_dimensions.html ├── docs_src ├── README.md ├── book.toml └── src │ ├── SUMMARY.md │ ├── complex_numbers.md │ ├── contribution.md │ ├── introduction.md │ ├── tin.md │ └── tube │ ├── advanced_field_construction.md │ ├── basics.md │ └── dynamic_dimensions.md ├── logo.PNG ├── pyproject.toml ├── requirements.txt ├── setup.py ├── src └── stannum │ ├── __init__.py │ ├── auxiliary.py │ ├── tin.py │ ├── tube.py │ └── utils.py └── tests ├── test_set_extra_kernel_args.py ├── test_tin ├── test_complex_grad.py ├── test_multi_kernel_grad.py └── test_simple_grad.py └── test_tube ├── test_backward.py ├── test_batch.py ├── test_dynamic_shape.py └── test_forward.py /.github/workflows/build_publish_stannum.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/.github/workflows/build_publish_stannum.yaml -------------------------------------------------------------------------------- /.github/workflows/run_tests.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/.github/workflows/run_tests.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/README.md -------------------------------------------------------------------------------- /benchmark/performance_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/benchmark/performance_benchmark.py -------------------------------------------------------------------------------- /docs/.nojekyll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/.nojekyll -------------------------------------------------------------------------------- /docs/404.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/404.html -------------------------------------------------------------------------------- /docs/CNAME: -------------------------------------------------------------------------------- 1 | fengliang.io 2 | -------------------------------------------------------------------------------- /docs/FontAwesome/css/font-awesome.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/FontAwesome/css/font-awesome.css -------------------------------------------------------------------------------- /docs/FontAwesome/fonts/FontAwesome.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/FontAwesome/fonts/FontAwesome.ttf -------------------------------------------------------------------------------- /docs/FontAwesome/fonts/fontawesome-webfont.eot: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/FontAwesome/fonts/fontawesome-webfont.eot -------------------------------------------------------------------------------- /docs/FontAwesome/fonts/fontawesome-webfont.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/FontAwesome/fonts/fontawesome-webfont.svg -------------------------------------------------------------------------------- /docs/FontAwesome/fonts/fontawesome-webfont.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/FontAwesome/fonts/fontawesome-webfont.ttf -------------------------------------------------------------------------------- /docs/FontAwesome/fonts/fontawesome-webfont.woff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/FontAwesome/fonts/fontawesome-webfont.woff -------------------------------------------------------------------------------- /docs/FontAwesome/fonts/fontawesome-webfont.woff2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/FontAwesome/fonts/fontawesome-webfont.woff2 -------------------------------------------------------------------------------- /docs/ayu-highlight.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/ayu-highlight.css -------------------------------------------------------------------------------- /docs/book.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/book.js -------------------------------------------------------------------------------- /docs/clipboard.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/clipboard.min.js -------------------------------------------------------------------------------- /docs/complex_numbers.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/complex_numbers.html -------------------------------------------------------------------------------- /docs/contribution.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/contribution.html -------------------------------------------------------------------------------- /docs/css/chrome.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/css/chrome.css -------------------------------------------------------------------------------- /docs/css/general.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/css/general.css -------------------------------------------------------------------------------- /docs/css/print.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/css/print.css -------------------------------------------------------------------------------- /docs/css/variables.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/css/variables.css -------------------------------------------------------------------------------- /docs/elasticlunr.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/elasticlunr.min.js -------------------------------------------------------------------------------- /docs/favicon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/favicon.png -------------------------------------------------------------------------------- /docs/favicon.svg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/favicon.svg -------------------------------------------------------------------------------- /docs/fonts/OPEN-SANS-LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/OPEN-SANS-LICENSE.txt -------------------------------------------------------------------------------- /docs/fonts/SOURCE-CODE-PRO-LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/SOURCE-CODE-PRO-LICENSE.txt -------------------------------------------------------------------------------- /docs/fonts/fonts.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/fonts.css -------------------------------------------------------------------------------- /docs/fonts/open-sans-v17-all-charsets-300.woff2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/open-sans-v17-all-charsets-300.woff2 -------------------------------------------------------------------------------- /docs/fonts/open-sans-v17-all-charsets-300italic.woff2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/open-sans-v17-all-charsets-300italic.woff2 -------------------------------------------------------------------------------- /docs/fonts/open-sans-v17-all-charsets-600.woff2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/open-sans-v17-all-charsets-600.woff2 -------------------------------------------------------------------------------- /docs/fonts/open-sans-v17-all-charsets-600italic.woff2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/open-sans-v17-all-charsets-600italic.woff2 -------------------------------------------------------------------------------- /docs/fonts/open-sans-v17-all-charsets-700.woff2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/open-sans-v17-all-charsets-700.woff2 -------------------------------------------------------------------------------- /docs/fonts/open-sans-v17-all-charsets-700italic.woff2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/open-sans-v17-all-charsets-700italic.woff2 -------------------------------------------------------------------------------- /docs/fonts/open-sans-v17-all-charsets-800.woff2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/open-sans-v17-all-charsets-800.woff2 -------------------------------------------------------------------------------- /docs/fonts/open-sans-v17-all-charsets-800italic.woff2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/open-sans-v17-all-charsets-800italic.woff2 -------------------------------------------------------------------------------- /docs/fonts/open-sans-v17-all-charsets-italic.woff2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/open-sans-v17-all-charsets-italic.woff2 -------------------------------------------------------------------------------- /docs/fonts/open-sans-v17-all-charsets-regular.woff2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/open-sans-v17-all-charsets-regular.woff2 -------------------------------------------------------------------------------- /docs/fonts/source-code-pro-v11-all-charsets-500.woff2: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/fonts/source-code-pro-v11-all-charsets-500.woff2 -------------------------------------------------------------------------------- /docs/highlight.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/highlight.css -------------------------------------------------------------------------------- /docs/highlight.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/highlight.js -------------------------------------------------------------------------------- /docs/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/index.html -------------------------------------------------------------------------------- /docs/introduction.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/introduction.html -------------------------------------------------------------------------------- /docs/mark.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/mark.min.js -------------------------------------------------------------------------------- /docs/print.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/print.html -------------------------------------------------------------------------------- /docs/searcher.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/searcher.js -------------------------------------------------------------------------------- /docs/searchindex.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/searchindex.js -------------------------------------------------------------------------------- /docs/searchindex.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/searchindex.json -------------------------------------------------------------------------------- /docs/tin.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/tin.html -------------------------------------------------------------------------------- /docs/tomorrow-night.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/tomorrow-night.css -------------------------------------------------------------------------------- /docs/tube/advanced_field_construction.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/tube/advanced_field_construction.html -------------------------------------------------------------------------------- /docs/tube/basics.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/tube/basics.html -------------------------------------------------------------------------------- /docs/tube/dynamic_dimensions.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs/tube/dynamic_dimensions.html -------------------------------------------------------------------------------- /docs_src/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs_src/README.md -------------------------------------------------------------------------------- /docs_src/book.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs_src/book.toml -------------------------------------------------------------------------------- /docs_src/src/SUMMARY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs_src/src/SUMMARY.md -------------------------------------------------------------------------------- /docs_src/src/complex_numbers.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs_src/src/complex_numbers.md -------------------------------------------------------------------------------- /docs_src/src/contribution.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs_src/src/contribution.md -------------------------------------------------------------------------------- /docs_src/src/introduction.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs_src/src/introduction.md -------------------------------------------------------------------------------- /docs_src/src/tin.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs_src/src/tin.md -------------------------------------------------------------------------------- /docs_src/src/tube/advanced_field_construction.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs_src/src/tube/advanced_field_construction.md -------------------------------------------------------------------------------- /docs_src/src/tube/basics.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs_src/src/tube/basics.md -------------------------------------------------------------------------------- /docs_src/src/tube/dynamic_dimensions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/docs_src/src/tube/dynamic_dimensions.md -------------------------------------------------------------------------------- /logo.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/logo.PNG -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | taichi 2 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/setup.py -------------------------------------------------------------------------------- /src/stannum/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/src/stannum/__init__.py -------------------------------------------------------------------------------- /src/stannum/auxiliary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/src/stannum/auxiliary.py -------------------------------------------------------------------------------- /src/stannum/tin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/src/stannum/tin.py -------------------------------------------------------------------------------- /src/stannum/tube.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/src/stannum/tube.py -------------------------------------------------------------------------------- /src/stannum/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/src/stannum/utils.py -------------------------------------------------------------------------------- /tests/test_set_extra_kernel_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/tests/test_set_extra_kernel_args.py -------------------------------------------------------------------------------- /tests/test_tin/test_complex_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/tests/test_tin/test_complex_grad.py -------------------------------------------------------------------------------- /tests/test_tin/test_multi_kernel_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/tests/test_tin/test_multi_kernel_grad.py -------------------------------------------------------------------------------- /tests/test_tin/test_simple_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/tests/test_tin/test_simple_grad.py -------------------------------------------------------------------------------- /tests/test_tube/test_backward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/tests/test_tube/test_backward.py -------------------------------------------------------------------------------- /tests/test_tube/test_batch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/tests/test_tube/test_batch.py -------------------------------------------------------------------------------- /tests/test_tube/test_dynamic_shape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/tests/test_tube/test_dynamic_shape.py -------------------------------------------------------------------------------- /tests/test_tube/test_forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ifsheldon/stannum/HEAD/tests/test_tube/test_forward.py --------------------------------------------------------------------------------