PyPI page
Home page
Author:
None
License:
MIT
Summary:
Monkey-patch PyTorch with optimized CuTile kernels for training (forward + backward)
Latest version:
0.1.3
Required dependencies:
cuda-tile
|
liger-kernel
|
torch
|
transformers
Optional dependencies:
liger-kernel
|
matplotlib
|
pytest
|
pytest-benchmark
|
ruff
Downloads last day:
0
Downloads last week:
9
Downloads last month:
48