Packages

GPU tensor backend for Nx via Vulkan compute. Pure-Rust path (vulkano) plus C++ spirit path. Validated on Axon training, Scholar linear regression, and the eXMC NUTS sampler. Works on Linux + FreeBSD NVIDIA where CUDA does not exist.

Current section

Files

Jump to
nx_vulkan priv shaders reduce_axis.spv
Raw

priv/shaders/reduce_axis.spv

Contents for binary files are not shown.