jax-cce

0.1.0

Memory-efficient fused linear cross-entropy loss for JAX. Computes cross_entropy(x @ w.T, labels) without materializing the full logits tensor.

License

Apache-2.0

Permissive

A permissive license whose main conditions require preservation of copyright and license notices. Contributors provide an express grant of patent rights. Licensed works, modifications, and larger works may be distributed under different terms and without source code.

Published

March 17, 2026

7d ago

Package Registry

PyPI Source Homepage

README badge Customize →

License Sources Match

Source	License	Class
Licensie (detected)	Apache-2.0	Permissive
PyPI (reported)	Apache-2.0	Permissive

Loading dependencies…

License File

Added Removed Expected

Versions

1 version

Version	License	Published	Status
0.1.0 Latest Viewing	Apache-2.0	Mar 17, 2026	Scanned