https://pypi.org/project/jax-cce/ Recent updates to the Python Package Index for jax-cce en https://pypi.org/project/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. Tue, 17 Mar 2026 08:01:17 GMT