-
Notifications
You must be signed in to change notification settings - Fork 3.7k
All issues
Issue creation is restricted in this repository
Issues
is:issue state:open
is:issue state:open
Search results
Reverse-mode double differentiation of saturated tanh returns 0 instead of the correct second derivative
bugSomething isn't workingSomething isn't workingStatus: Open.#40315聽In jax-ml/jax;jit lower depends on process trace history
bugSomething isn't workingSomething isn't workingStatus: Open.#40312聽In jax-ml/jax;Aval repr is lossy.
bugSomething isn't workingSomething isn't workingStatus: Open.#40288聽In jax-ml/jax;transfer_guard and jax.device_put(..., device=cpu_device)
bugSomething isn't workingSomething isn't workingStatus: Open.#40285聽In jax-ml/jax;Why the input_output_alias disappear?
bugSomething isn't workingSomething isn't workingStatus: Open.#40280聽In jax-ml/jax;Programmatic profiler results
enhancementNew feature or requestNew feature or requestStatus: Open.#40258聽In jax-ml/jax;All-NaN gradients from preconditioned
jax.scipy.sparse.linalg.cgfor small cotangent normsbugSomething isn't workingSomething isn't workingStatus: Open.#40254聽In jax-ml/jax;Shape-dependent XLA/Triton failure for jvp(grad) through CNN
bugSomething isn't workingSomething isn't workingStatus: Open.#40237聽In jax-ml/jax;Support GMEM ranks above two in the tiled
cp.asyncGMEM鈫扴MEM copy (pre-Hopper)enhancementNew feature or requestNew feature or requestStatus: Open.#40201聽In jax-ml/jax;[Pallas TPU] 4-KiB-aligned dynamic HBM-to-HOST DMA is incorrectly rejected as not 4-KiB aligned
bugSomething isn't workingSomething isn't workingStatus: Open.#40200聽In jax-ml/jax;jnp.linalg.svdgradients returnnanfor intermediate overflowbugSomething isn't workingSomething isn't workingStatus: Open.#40185聽In jax-ml/jax;[Mosaic GPU] Support collective tcgen05.commit arrivals in multidimensional clusters
enhancementNew feature or requestNew feature or requestStatus: Open.#40165聽In jax-ml/jax;
