PyPI page
Home page
Author:
Shizheng Wen
Summary:
Fused Triton kernels for the slice/deslice bottleneck of physics-attention, and the Transolver layer they accelerate.
Latest version:
0.3.0
Required dependencies:
einops
|
torch
|
triton
Optional dependencies:
pytest
Downloads last day:
4
Downloads last week:
6
Downloads last month:
248
7-day smoothing