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.0.dev20260706151306
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:
1,474
Downloads last week:
12,442
Downloads last month:
25,562