A custom kernel for PyTorch.
pip install __REPO_ID__import torch from __KERNEL_NAME_NORMALIZED__ import __KERNEL_NAME_NORMALIZED__ # Create input tensor x = torch.randn(1024, 1024, device="cuda") # Run kernel result = __KERNEL_NAME_NORMALIZED__(x)nix develop nix run .#build-and-copynix develop .#test pytest tests/uv run example.pyApache 2.0