piwheels

cut-cross-entropy

Code for cut cross entropy, a memory efficient implementation of linear-cross-entropy loss.

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install cut-cross-entropy

Dependencies

• torch
• triton

Releases

Version	Released	Bullseye Python 3.9	Bookworm Python 3.11	Trixie Python 3.13	Files
25.1.1	2025-01-07
cut_cross_entropy-25.1.1-py3-none-any.whl (22 KB)
24.12.3	2024-12-24
cut_cross_entropy-24.12.3-py3-none-any.whl (22 KB)
24.12.2	2024-12-10
cut_cross_entropy-24.12.2-py3-none-any.whl (22 KB)
24.12.1	2024-12-04
cut_cross_entropy-24.12.1-py3-none-any.whl (22 KB)
24.11.4	2024-11-17
cut_cross_entropy-24.11.4-py3-none-any.whl (22 KB)
24.11.3	2024-11-17
cut_cross_entropy-24.11.3-py3-none-any.whl (22 KB)

Issues with this package?

• Search issues for this package
• Package or version missing? Open a new issue
• Something else? Open a new issue