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 permitted provided that the following...
Summary:
Efficient Triton kernels for LLM Training
Latest version:
0.5.2
Required dependencies:
torch
|
triton
Optional dependencies:
black
|
datasets
|
flake8
|
isort
|
matplotlib
|
pytest
|
pytest-rerunfailures
|
pytest-xdist
|
seaborn
|
setuptools-scm
|
torch
|
torchvision
|
transformers
|
triton
|
trl
Downloads last day:
2,021
Downloads last week:
17,731
Downloads last month:
587,645