Instructions to use replicate/trimul-gpumode with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Kernels
How to use replicate/trimul-gpumode with Kernels:
# !pip install kernels from kernels import get_kernel # a version (or an explicit revision) is required; see the "Files and versions" tab for the available ones kernel = get_kernel("replicate/trimul-gpumode", version=1) - Notebooks
- Google Colab
- Kaggle
File size: 280 Bytes
cd40f46 | 1 2 3 4 5 6 7 | from .triton_a100 import kernel_a100
from .triton_h100 import kernel_h100
from .triton_b200 import kernel_b200
from .trimul_mi300 import kernel_mi300
from .trimul_global import kernel_global
__all__ = ["kernel_a100", "kernel_h100", "kernel_b200", "kernel_mi300", "kernel_global"] |