PyPI page
Home page
Author:
None
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.1
Required dependencies:
torch
|
triton
Optional dependencies:
cuda-tile
|
datasets
|
matplotlib
|
mkdocs-material
|
nvidia-cutlass-dsl
|
prek
|
pytest
|
pytest-asyncio
|
pytest-cov
|
pytest-rerunfailures
|
pytest-xdist
|
ruff
|
seaborn
|
torchvision
|
transformers
Downloads last day:
22,208
Downloads last week:
140,271
Downloads last month:
587,745