Module hardware

Module hardware 

Source
Expand description

Hardware abstraction layer

Provides unified interface for different hardware backends:

  • CUDA (NVIDIA GPUs)
  • ROCm (AMD GPUs)
  • Metal (Apple Silicon)
  • TPU (Google TPUs)
  • CPU with SIMD (AVX, NEON)

Structs§

HardwareDevice
Hardware device information

Enums§

ElementwiseOp
HardwareBackend
Hardware backend type

Traits§

HardwareOps
Hardware-accelerated operations trait

Functions§

list_devices
List available devices