API Reference nx_vulkan v#0.1.0

Copy Markdown View Source

Modules

Nx tensor backend on Vulkan compute.

Nx.Backend implementation on top of the Vulkan compute primitives.

Catalog of family specs for templated chain-shader synthesis.

Path A.2 v2 (partial) — Nx.Defn.Compiler that auto-detects fusable elementwise chains and dispatches Nx.Vulkan.fused_chain/3 instead of N separate shader calls.

Named fused kernels for MCMC hot paths.

Path A.2 — compile-time fusion of elementwise op chains.

Rustler NIF bindings for the Vulkan compute backend.

Rustler NIF for the pure-Rust (vulkano) compute backend.

Long-lived per-machine GPU node. A named GenServer that owns the spirit VkPipelineCache, the persistent buffer registry, and the watchdog/timeout layer. Clients submit work via with_node/2 (or the lower-level exec/2); the node serializes execution and reports timeouts/dead-server/etc. as error tuples.

Phase 2 W5 — disk persistence for spirit's VkPipelineCache.

Phase 1 — templated GLSL shader synthesis.

Per-family hooks for the templated chain shader.

Phase 1 — end-to-end shader synthesis driver.

Pure-Rust (vulkano) Nx.Backend implementation. Sibling of Nx.Vulkan.Backend (C++ spirit-backed); same compute fabric, different memory-management story.