Runtime ND4J Backend type selection


I have a use case where I would like to use both CPU and CUDA backends. Backend type for a specific job should be decided based on job type, job size, available memory for the backends and a couple of other factors.
Is it possible to tell ND4J at runtime which backend to be used for the next INDArray declaration?

1 Like