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 Report package

Current section

Files

Jump to

Readme 3 Versions 3 Dependencies Dependants Files Activity

nx_vulkan .formatter.exs

Raw

97 files

docs
examples
lib
livebooks
native
priv
.formatter.exs
CHANGELOG.md
LICENSE
README.md
mix.exs

.formatter.exs

# Used by "mix format"
[
  inputs: ["{mix,.formatter}.exs", "{config,lib,test}/**/*.{ex,exs}"]
]

Search filters

Type any of these into the search box. They can be combined with free text which searches package names and descriptions.

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

Wrap values in double quotes to include spaces, e.g. name:"my package".