PyPI page
Home page
Author:
None
Summary:
Code for cut cross entropy, a memory efficient implementation of linear-cross-entropy loss.
Latest version:
24.11.4
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:
5,954
Downloads last week:
22,474
Downloads last month:
22,617