-
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
- Status: Open.#40260 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→SMEM 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;- Status: Open.#40150 In jax-ml/jax;
Repeated XLA compiles of jnp.argsort on the CUDA backend crashes
bugSomething isn't workingSomething isn't workingStatus: Open.#40142 In jax-ml/jax;splash attention:
dsinksgradient loses precision by rounding before the reductionbugSomething isn't workingSomething isn't workingStatus: Open.#40128 In jax-ml/jax;- Status: Open.#40101 In jax-ml/jax;