Factual package intelligence from PyPI

jax-cce 0.1.0 package intelligence

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

pip install jax-cce

Current version
0.1.0
Python requirement
>=3.11
License
Apache-2.0
Distribution type
Pure Python wheel
Release files
2 (1 wheels, 1 source)
Download size
26.6 KiB
Release history
1 releases with files
Median release cadence
Not enough dated releases

jax-cce dependencies

PyPI declares 1 unique dependency rule for this release. Environment markers are shown when supplied by the project.

  1. jax

Compatibility and release files

jax-cce publishes 1 wheel and 1 source archive for version 0.1.0. Wheel platform tags: any.

No version-specific Python classifiers are declared.

Release activity and sources

PyPI lists 1 release with files. The first dated release is ; 1 release falls within the 365 days preceding the latest dated release. The current release files were uploaded on .