PyPI page
Home page
Author:
Patrick Toulmé
Summary:
Python DSL for writing PTX kernels, callable from JAX, PyTorch, and torch.compile
Latest version:
0.1.1
Optional dependencies:
cuda-python
|
jax
|
mkdocs-material
|
mkdocstrings
|
numpy
|
pytest
|
pytest-xdist
|
ruff
|
torch
Downloads last day:
58
Downloads last week:
465
Downloads last month:
1,119