ASTRA KernelKit documentation

ASTRA KernelKit is an all-Python tomographic reconstruction package, leveraging the GPU-accelerated tomographic projection kernels of the ASTRA Toolbox using CuPy.

• Develop, explore and debug CT algorithms and X-ray projectors with the Python ecosystem.

• Leverage CUDA features, such as CUDA graphs or unified memory, for maximum performance.

• Kernels are compiled during the script with NVRTC and Jinja2. Hence, no complicated build system, and yet easy customization of kernels.

• Easy to install and integrates well with PyTorch and other GPU frameworks!

Try ASTRA KernelKit yourself, visit Installation.

Note

For ASTRA Toolbox visit https://astra-toolbox.readthedocs.io/en/latest/.