Hugging Face's logo

mohitsha
/
aiter_aiter_operator

Model card Files Community
YAML Metadata Warning: empty or missing yaml metadata in repo card (https://huggingface.co/docs/hub/model-cards#model-card-metadata)
import torch
from kernels import get_kernel
# Download optimized kernels from the Hugging Face hub
ops = get_kernel("mohitsha/aiter_aiter_operator")
# Random tensor
x = torch.randn((10, 10), dtype=torch.bfloat16, device="cuda")
y = torch.randn((10, 10), dtype=torch.bfloat16, device="cuda")
out = ops.div(y, x)
print(out)
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Collection including mohitsha/aiter_aiter_operator