Requirements

Hardware

• NVIDIA GPU with CUDA support (for GPU-specific tests and benchmarks).

Software

• Python 3.10 or later

• PyTorch (with CUDA support)

• Triton

• CuPy (cupy-cuda12x)