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 kinetic_energy.spv
Raw

priv/shaders/kinetic_energy.spv

Contents for binary files are not shown.