Instructions to use replicate/deep-gemm with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Kernels
How to use replicate/deep-gemm 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/deep-gemm", version=1) - Notebooks
- Google Colab
- Kaggle
Download torch-ext/deep_gemm/jit/__init__.py from replicate/deep-gemm: direct link, hf CLI and curl.
- Browser
- Download file 119 Bytes
-
https://huggingface.co/replicate/deep-gemm/resolve/main/torch-ext/deep_gemm/jit/__init__.py
- Command line
-
hf download hf://replicate/deep-gemm/torch-ext/deep_gemm/jit/__init__.py
-
curl -L -o __init__.py https://huggingface.co/replicate/deep-gemm/resolve/main/torch-ext/deep_gemm/jit/__init__.py
119 Bytes
| from .compiler import get_nvcc_compiler, build | |
| from .template import cpp_format, generate | |
| from .runtime import Runtime | |