PyPI page
Home page
Author:
None
Summary:
Code for cut cross entropy, a memory efficient implementation of linear-cross-entropy loss.
Latest version:
25.1.1
Required dependencies:
deepspeed
|
torch
|
triton
Optional dependencies:
accelerate
|
build
|
cut-cross-entropy
|
datasets
|
fire
|
huggingface_hub
|
pandas
|
pre-commit
|
pytest
|
tqdm
|
transformers
|
twine
Downloads last day:
13,105
Downloads last week:
88,243
Downloads last month:
370,072