gpu_device: GPU device

View source: R/core.R

gpu_deviceR Documentation

GPU device

Description

A GPU compute device from MLDataDevices.jl for the device argument of train().

Usage

gpu_device(...)

Arguments

...

arguments passed to the Julia function (e.g. force).

Value

a Julia device object

See Also

cpu_device(), reactant_device(), train()


NeuralEstimators documentation built on Sept. 10, 2026, 1:12 a.m.