Skip to content

Issues: jax-ml/jax

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

1,508 Open 4,491 Closed
1,508 Open 4,491 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

Results mismatch between different GEMM algorithms on AMD Instinct accelerator bug Something isn't working
#26846 opened Feb 28, 2025 by ffrancesco94
[Jax GPU] Unexpected difference when results should be equal bug Something isn't working
#26836 opened Feb 28, 2025 by hanzhi713
Fix tokens for arrays with no local shards bug Something isn't working
#26815 opened Feb 27, 2025 by emilyfertig
2
[Pallas] python for loop in kernel function is unbearably slow compared to fori_loop bug Something isn't working
#26812 opened Feb 27, 2025 by ysngshn
5
expm yields wrong values even with max_squarings set to high values bug Something isn't working
#26802 opened Feb 27, 2025 by ymardoukhi
@jakevdp
1
Implement custom_linear_solve and eigh primtive rules for Taylor series propagation in jax.experimental.jet enhancement New feature or request
#26801 opened Feb 27, 2025 by yachmena
1
Improve debug_nan in shard_map to pinpoint exact line of NaN occurrence enhancement New feature or request
#26796 opened Feb 27, 2025 by Stella-S-Yan
Deterministic execution produces different results for single vs. batch input on both CPU and GPU bug Something isn't working
#26795 opened Feb 27, 2025 by jashvira
2
Warning generated from jit + device-to-host + tokens bug Something isn't working
#26783 opened Feb 26, 2025 by kcdodd
1
xla_extension.so: cannot enable executable stack as shared object requires: Invalid argument bug Something isn't working
#26781 opened Feb 26, 2025 by DSilva27
3
3
Jax compilation hangs indefinitely bug Something isn't working
#26767 opened Feb 26, 2025 by n-gao
__array_priority__ is not checked by JAX for downstream custom arrays bug Something isn't working
#26764 opened Feb 26, 2025 by gautierronan
@jakevdp
6
shard_map not working with symbolic_zero in custom_jvp enhancement New feature or request
#26763 opened Feb 26, 2025 by n-gao
@mattjj
1
Termination timeout error bug Something isn't working
#26758 opened Feb 26, 2025 by andrewilyas
1
AttributeError: module 'jaxlib.xla_client' has no attribute 'register_custom_type_id_handler'. bug Something isn't working
#26750 opened Feb 25, 2025 by rohandekate10
@justinjfu
6
Matrix multiply incorrect for large tensors bug Something isn't working
#26689 opened Feb 23, 2025 by unalmis
1
Pallas matmul doesn't match jnp.matmul with interpret=True bug Something isn't working
#26684 opened Feb 23, 2025 by PaulScemama
@justinjfu
1
Feature request: Pallas implementation of NSA Attention (by deepseek) enhancement New feature or request
#26681 opened Feb 22, 2025 by OhadRubin
Tracking: MPMD/pipeline parallelism in eager-mode McJAX enhancement New feature or request
#26645 opened Feb 20, 2025 by emilyfertig
Plumb custom layouts through array creation enhancement New feature or request
#26642 opened Feb 20, 2025 by emilyfertig
@yashk2810
Unexpectedly high memory consumption in jax.lax.scan bug Something isn't working
#26639 opened Feb 20, 2025 by dbezgin
jax.scipy.linalg.expm broken under shard_map bug Something isn't working
#26621 opened Feb 19, 2025 by PhilipVinc
@mattjj
Scan leads to much higher memory usage than an unrolled loop bug Something isn't working
#26618 opened Feb 19, 2025 by marcelroed
pmap-ing slower with new CPU runtime bug Something isn't working
#26616 opened Feb 19, 2025 by lockwo
2
__repr__ or __str__ method for jax._src.config.Config objects enhancement New feature or request
#26608 opened Feb 19, 2025 by ZacCranko
ProTip! Find all open issues with in progress development work with linked:pr.