PyPI page
Home page
License:
BSD 2-CLAUSE LICENSE
Copyright 2024 LinkedIn Corporation
All Rights Reserved.
Redistribution and use in source and binary forms, with or
without modification, are perm...
Summary:
Efficient Triton kernels for LLM Training
Latest version:
0.8.5
Required dependencies:
torch
|
triton
Optional dependencies:
apache-tvm-ffi
|
cuda-tile
|
datasets
|
liger-cute-kernels
|
matplotlib
|
mkdocs-material
|
nvidia-cutlass-dsl
|
prek
|
pytest
|
pytest-asyncio
|
pytest-cov
|
pytest-rerunfailures
|
pytest-xdist
|
ruff
|
seaborn
|
torchvision
|
transformers
Downloads last day:
29,220
Downloads last week:
162,882
Downloads last month:
448,508
7-day smoothing