-
Notifications
You must be signed in to change notification settings - Fork 3.7k
Issues
is:issue state:open
is:issue state:open
Issue creation is restricted in this repository
Search results
Make local_to_global_shape public
enhancementNew feature or requestNew feature or requestStatus: Open.#38889 In jax-ml/jax;jnp.max uses input precision for VJP reduction
bugSomething isn't workingSomething isn't workingStatus: Open.#38885 In jax-ml/jax;- Status: Open.#38835 In jax-ml/jax;
- Status: Open.#38832 In jax-ml/jax;
Feature request: add differentiation rule for jax.scipy.linalg.schur (NotImplementedError on jax.grad)
enhancementNew feature or requestNew feature or requestStatus: Open.#38831 In jax-ml/jax;register_hlo_module_transformationcan deadlock on TPUCompileWithTopologybecause the Python transform callback needs the GILbugSomething isn't workingSomething isn't workingStatus: Open.#38829 In jax-ml/jax;- Status: Open.#38828 In jax-ml/jax;
jnp.kron self-composition causes XLA static extent overflow with opaque internal error
bugSomething isn't workingSomething isn't workingStatus: Open.#38826 In jax-ml/jax;jax.scipy.linalg.inv and cholesky raise confusing internal KeyError for unsupported dtypes (float16, bfloat16)
bugSomething isn't workingSomething isn't workingStatus: Open.#38825 In jax-ml/jax;jax.nn.scaled_matmul raises NotImplementedError on CPU with no graceful error
bugSomething isn't workingSomething isn't workingStatus: Open.#38813 In jax-ml/jax;GPU scatter-add (
x.at[idx].add(y)) into a small array up to ~2000x slower in jaxlib 0.10.2 vs 0.10.1bugSomething isn't workingSomething isn't workingStatus: Open.#38806 In jax-ml/jax;jax.vmap produces different results from eager for a function involving jnp.round, jnp.inner, jax.lax.collapse, jnp.vecdot, and jnp.remainder
bugSomething isn't workingSomething isn't workingStatus: Open.#38768 In jax-ml/jax;