Avatar for Mehul Totala from gravatar.com

Mehul Totala

Username    Coffeempty
Date joined   Joined

1 project

jax-cce

Last released

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

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page