jax4j / NablaTensor

Docs / jax4j-core / com.nablatensor.jax4j.tensor

enum

Backend

User-facing backend selector for jit/function builders.

View source ↗Full module reference on javadoc.io ↗

Enum constants

AUTO
CPU
CUDA
ROCM
VULKAN
WGPU

Methods

static Backend parse(String value)