voiage.core.gpu_acceleration

GPU acceleration utilities for Value of Information analysis.

Classes

• GPUAcceleratedEVPI GPU-accelerated Expected Value of Perfect Information (EVPI) calculator.

Functions

• get_gpu_backend Get the available GPU backend.
• is_gpu_available Check if any GPU backend is available.
• array_to_gpu Transfer a NumPy array to GPU memory using the specified or default backend.
• array_to_cpu Transfer an array from GPU memory back to CPU (NumPy).
• gpu_jit_compile JIT compile a function using the specified or default GPU backend.
• gpu_vectorize Vectorize a function using the specified or default GPU backend.
• gpu_parallelize Parallelize a function across multiple GPUs using the specified or default backend.
• example_gpu_acceleration Use GPU acceleration utilities.