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.6.4
Required dependencies:
torch
|
triton
Optional dependencies:
datasets
|
matplotlib
|
mkdocs-material
|
pytest
|
pytest-asyncio
|
pytest-cov
|
pytest-rerunfailures
|
pytest-xdist
|
ruff
|
seaborn
|
torchvision
|
transformers
Downloads last day:
18,377
Downloads last week:
101,782
Downloads last month:
535,936