cut-cross-entropy

View on PyPIReverse Dependencies (2)

24.11.4 cut_cross_entropy-24.11.4-py3-none-any.whl

Wheel Details

Project: cut-cross-entropy
Version: 24.11.4
Filename: cut_cross_entropy-24.11.4-py3-none-any.whl
Download: [link]
Size: 22487
MD5: 58a15e3c3232e4c38e7217cf75a4bf9a
SHA256: 00f8799dc6623bf4fdd2e35c2b2bd9bdbafc66d3342217cd31da5f87f7eabed6
Uploaded: 2024-11-17 21:16:18 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: cut-cross-entropy
Version: 24.11.4
Summary: Code for cut cross entropy, a memory efficient implementation of linear-cross-entropy loss.
Requires-Python: >=3.10
Requires-Dist: torch
Requires-Dist: triton
Requires-Dist: cut-cross-entropy[transformers]; extra == "all"
Requires-Dist: accelerate (>=0.34.2); extra == "all"
Requires-Dist: datasets (>=3.1.0); extra == "all"
Requires-Dist: huggingface-hub (>=0.26.2); extra == "all"
Requires-Dist: pandas; extra == "all"
Requires-Dist: fire; extra == "all"
Requires-Dist: tqdm; extra == "all"
Requires-Dist: deepspeed (>=0.15.1); platform_system != "Darwin" and extra == "all"
Requires-Dist: cut-cross-entropy[all,test]; extra == "dev"
Requires-Dist: build; extra == "dev"
Requires-Dist: twine; extra == "dev"
Requires-Dist: pre-commit; extra == "dev"
Requires-Dist: pytest; extra == "test"
Requires-Dist: transformers (>=4.44.2); extra == "transformers"
Provides-Extra: all
Provides-Extra: dev
Provides-Extra: test
Provides-Extra: transformers
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 7987 characters]

WHEEL

Wheel-Version: 1.0
Generator: setuptools (75.5.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
cut_cross_entropy/__init__.py sha256=F3K5VdKKXP92vzz-C_gXPn1Y0ia9OADglgfbAzd2xrM 326
cut_cross_entropy/cce.py sha256=tCqocDb5BEtSP0RlWINNt0vI0tD7skBL0FfJemHtIGg 5339
cut_cross_entropy/cce_backward.py sha256=Onf2uiLG2BqzfcXzC6IcEXOow303Jnp9p4MHb-edzis 9147
cut_cross_entropy/cce_lse_forward.py sha256=VJp8mrpN0xCu1IOq5VdQHfW9bX1ZsacVnVrukzq9a84 6086
cut_cross_entropy/constants.py sha256=LcLPytUgFUR98ED3WwKEE8nivhVESJ5wkWLrUXKeysc 80
cut_cross_entropy/doc.py sha256=_H4o7CMrPj0aVOUo4Rk-cQZfaExxOhaKXgAXfM6bLMs 1626
cut_cross_entropy/indexed_dot.py sha256=y0xXJJqbRc0rIGzGsotAp-_rNQBCBXdr50a_qcbRMBU 3506
cut_cross_entropy/linear_cross_entropy.py sha256=mSllz6S2DCR1Id5_imqGFBKi_KRGiCIeKLMvQ4ZMG94 3319
cut_cross_entropy/tl_autotune.py sha256=9rQwBNgvFyR6mWv1IQCTMB7jH2_coaQRIRCq9aiXUSM 15151
cut_cross_entropy/tl_utils.py sha256=F9ukF0zkf4iyv2DfYmVq802lVUZQMNgD-v8votVyc_M 1312
cut_cross_entropy/torch_compile.py sha256=J0JKocvE2WO0TNffGBMJI8tTGV8556OG13kKan4EC1M 1961
cut_cross_entropy/utils.py sha256=u-90THCjRRwQe1NagjUQmdKL6b2Fx4mxlc3ILvhkLDE 1941
cut_cross_entropy-24.11.4.dist-info/LICENSE sha256=yTPASVpJQ4IDvXmZGERUjmxoUyBRB_lcZowxeiwgIVU 2731
cut_cross_entropy-24.11.4.dist-info/METADATA sha256=VN0yp-y0qyYItb7Lhu9yhNjfN0cb1tJHT_G7bdSFuxE 9319
cut_cross_entropy-24.11.4.dist-info/WHEEL sha256=R06PA3UVYHThwHvxuRWMqaGcr-PuniXahwjmQRFMEkY 91
cut_cross_entropy-24.11.4.dist-info/top_level.txt sha256=dSZLLiYIfH2pCB4rvFQkuZqtFgxQFJvUhRCctLcJUYE 18
cut_cross_entropy-24.11.4.dist-info/RECORD

top_level.txt

cut_cross_entropy