kernels
The Kernel Hub allows Python libraries and applications to load compute kernels directly from the Hub. To support this kind of dynamic loading, Hub kernels differ from traditional Python kernel packages in that they are made to be:
- Portable: a kernel can be loaded from paths outside
PYTHONPATH. - Unique: multiple versions of the same kernel can be loaded in the same Python process.
- Compatible: kernels must support all recent versions of Python and the different PyTorch build configurations (various CUDA versions and C++ ABIs). Furthermore, older C library versions must be supported.
🚀 Quick Start
Install the kernels package with pip (requires torch>=2.5 and CUDA):
pip install kernels
Here is how you would use the activation kernels from the Hugging Face Hub:
import torch
from kernels import get_kernel
# Download optimized kernels from the Hugging Face hub
activation = get_kernel("kernels-community/activation")
# Random tensor
x = torch.randn((10, 10), dtype=torch.float16, device="cuda")
# Run the kernel
y = torch.empty_like(x)
activation.gelu_fast(y, x)
print(y)
You can search for kernels on the Hub.
📚 Documentation
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file kernels-0.4.0.tar.gz.
File metadata
- Download URL: kernels-0.4.0.tar.gz
- Upload date:
- Size: 13.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.0.1 CPython/3.12.8
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
7a4a4bd2d53c801ac58a5e8bcf09b8e2c76b9e234b56e07846625644864c87b1
|
|
| MD5 |
fe08495657d6dda4968ac7784a501b38
|
|
| BLAKE2b-256 |
efc9854605083625ec1270cecaf4ece1401d1fd1c2f816352ff690dbc6f43187
|
File details
Details for the file kernels-0.4.0-py3-none-any.whl.
File metadata
- Download URL: kernels-0.4.0-py3-none-any.whl
- Upload date:
- Size: 11.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.0.1 CPython/3.12.8
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
389af5c4a013179f3d60ae00c60173157cb59588335e2cca5ac4d60bf9aa4ad4
|
|
| MD5 |
4df44a1e449d34826c9bb1fb48eae116
|
|
| BLAKE2b-256 |
e33f9d438d6fdc2798d7bb21808a21dd27dc6651f0a92be4dd7e583f67d176e6
|
