#

cuda-kernel

Here are 2 public repositories matching this topic...

teddykoker / torchsort

Fast, differentiable sorting and ranking in PyTorch

pytorch sort ranking cuda-kernel

Updated Dec 29, 2023
Python

webis-de / pytorch-window-matmul

a custom CUDA kernel for windowed matrix multiplication

cuda pytorch cuda-kernel

Updated Jan 3, 2024
Python

Improve this page

Add a description, image, and links to the cuda-kernel topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the cuda-kernel topic, visit your repo's landing page and select "manage topics."