Skip to content

Issues: google/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,301 Open 3,856 Closed
1,301 Open 3,856 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

Need a Discord group for daily development communication enhancement New feature or request
#21269 opened May 17, 2024 by knightXun
1
Compilation takes way more time in new version bug Something isn't working needs info More information is required to diagnose & prioritize the issue. NVIDIA GPU Issues specific to NVIDIA GPUs XLA
#21250 opened May 15, 2024 by stergiosba
2
Checkify exception only reports error from one device (not all) bug Something isn't working
#21246 opened May 15, 2024 by billmark
7
Jax not recognizing GPU. bug Something isn't working
#21240 opened May 15, 2024 by charlie-guan
2
shard_map can cause misses in compilation_cache bug Something isn't working
#21236 opened May 15, 2024 by jaro-sevcik
@mattjj
1
Numerical differences between shardings in random algorithm bug Something isn't working
#21232 opened May 14, 2024 by shawnwang18
@froystig
3
shard map implemenation of pmap only works for the first signature bug Something isn't working
#21225 opened May 14, 2024 by jheek
1
Nested custom vjps do not correctly propagate cotangents bug Something isn't working
#21220 opened May 14, 2024 by lengstrom
@froystig @mattjj
1
Distributed training is stuck bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#21217 opened May 14, 2024 by syyxsxx
3
BF16 matmul slower than F32 matmul on T4 GPU bug Something isn't working
#21212 opened May 13, 2024 by sagelywizard
@hawkinsp
2
different behavior of rel_entr bug Something isn't working
#21192 opened May 12, 2024 by sh0416
@jakevdp
Pallas+scan: NotImplementedError when num_extensive is True bug Something isn't working
#21190 opened May 12, 2024 by AllanYangZhou
1
ComplexWarning in VJP when using complex matrix multiplication bug Something isn't working
#21188 opened May 11, 2024 by mathisgerdes
@mattjj
2
Compilation cache does not work with custom partitioning enhancement New feature or request
#21159 opened May 10, 2024 by jaro-sevcik
2
Strided indexing turns into gather enhancement New feature or request performance make things lean and fast
#21151 opened May 9, 2024 by jewillco
3
[pallas] Interpreter mismatch for masked OOB indexing bug Something isn't working
#21143 opened May 9, 2024 by oliverdutton
linalg.solve produces NaNs on GPU, but not on CPU bug Something isn't working
#21122 opened May 7, 2024 by fkaab
Crash in eval_jaxpr with 0.4.27 bug Something isn't working
#21116 opened May 7, 2024 by patrick-kidger
@yashk2810
17
Remaining deprecations for array API compliance enhancement New feature or request
#21088 opened May 6, 2024 by Micky774
2 of 13 tasks
ROCm 6.1, 7900 xtx: bfloat16 support not enabled? AMD GPU Issues pertaining to AMD GPUs (ROCM) bug Something isn't working
#21074 opened May 4, 2024 by brettkoonce
1
Persistent compilation cache does not work bug Something isn't working
#21067 opened May 3, 2024 by neel04
2
Unexpected speedup from wrapping function call in trivial jax.lax.cond statement bug Something isn't working
#21065 opened May 3, 2024 by TonyZhou729
dynamic config scope under jit doesn't change partitionable threefry behavior bug Something isn't working
#21061 opened May 3, 2024 by froystig
@froystig
Seeking guidance for landing spot of scipy.stats.levy_stable in Jax enhancement New feature or request
#21054 opened May 3, 2024 by tjhunter
buggy interaction: remat, automatic partitioning, and unsafe rbg-based RNGs bug Something isn't working
#21046 opened May 3, 2024 by froystig
@froystig @mattjj
ProTip! What’s not been updated in a month: updated:<2024-04-17.