Packages

GPU tensor backend for Nx via Vulkan compute (vulkano/Rust, native f32 and f64), forward and backward pass on-device. Validated on Axon training, Scholar linear regression, and the eXMC NUTS sampler. Works on Linux + FreeBSD NVIDIA where CUDA does not exist.

Retired package: Deprecated - Backward pass ran on the host: GPU training was ~250x slower than advertised. Results were correct; use 0.3.0 for training.

Current section

Readme

Jump to
Loading README...

Checksum

Dependency Config

mix.exs

rebar.config

Gleam

erlang.mk

Package Details

Downloads Last 30 days, 0.2.0
0 5 10 15 20

this version

36

yesterday

25

last 7 days

59

all time

162

Last Updated

Aug 04, 2026

License

Apache-2.0

Build Tools

mix

Publisher

octanix octanix

Owners