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:
14,080
Downloads last week:
91,648
Downloads last month:
367,975