entangle-jax

PyPI page
Home page
Author: Nardi Lam
Summary: A generic JAX primitive that is opaque to XLA reordering optimizations
Latest version: 0.1.1
Required dependencies: jax | jaxtyping

Downloads last day: 1
Downloads last week: 97
Downloads last month: 885
7-day smoothing