nx_vulkan

0.2.0

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.

HexDocs

Current section

3 Dependencies

Jump to

Readme 3 Versions 3 Dependencies 0 Dependants Files Activity

Dependencies of nx_vulkan

3 dependencies

Package	Requirement
jason	~> 1.4
nx	~> 0.13
rustler	~> 0.36.0

Checksum

Dependency Config

mix.exs

rebar.config

Gleam

erlang.mk

Package Details

Downloads Last 30 days, all versions

this version

yesterday

last 7 days

all time

162

Last Updated

Aug 04, 2026

License

Apache-2.0

Build Tools

mix

Publisher

octanix

Owners

octanix

Operator	Description	Example
name:	Match package name. Supports * wildcards and repo/package form	name:phx* or name:hexpm/phoenix
description:	Full-text search of package descriptions	description:auth
depends:	Packages depending on a given package. Supports repo:package form	depends:ecto or depends:hexpm:ecto
build_tool:	Filter by build tool	build_tool:mix
updated_after:	Packages updated after an ISO8601 datetime	updated_after:2025-01-01T00:00:00Z
extra:	Match custom metadata (key,value). Nested keys are separated by commas	extra:license,MIT