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.
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
Author
Label
Projects
Milestones
Assignee
Sort
Issues list
jax.profiler.trace
repeatedly fails to display entire trace
bug
#21295
opened May 18, 2024 by
jon-chuang
Improve precision of 32-bit Something isn't working
gammaln
?
bug
#21284
opened May 17, 2024 by
lucascolley
simple pallas kernel hangs when input size exceeds some threshold
bug
Something isn't working
pallas
Issues pertaining to Pallas (GPU or TPU)
#21282
opened May 17, 2024 by
zhixuan-lin
Need a Discord group for daily development communication
enhancement
New feature or request
#21269
opened May 17, 2024 by
knightXun
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
Checkify exception only reports error from one device (not all)
bug
Something isn't working
#21246
opened May 15, 2024 by
billmark
shard_map can cause misses in compilation_cache
bug
Something isn't working
#21236
opened May 15, 2024 by
jaro-sevcik
Numerical differences between shardings in random algorithm
bug
Something isn't working
#21232
opened May 14, 2024 by
shawnwang18
shard map implemenation of pmap only works for the first signature
bug
Something isn't working
#21225
opened May 14, 2024 by
jheek
Nested custom vjps do not correctly propagate cotangents
bug
Something isn't working
#21220
opened May 14, 2024 by
lengstrom
Distributed training is stuck
bug
Something isn't working
NVIDIA GPU
Issues specific to NVIDIA GPUs
#21217
opened May 14, 2024 by
syyxsxx
BF16 matmul slower than F32 matmul on T4 GPU
bug
Something isn't working
#21212
opened May 13, 2024 by
sagelywizard
Pallas+scan: NotImplementedError when num_extensive is True
bug
Something isn't working
#21190
opened May 12, 2024 by
AllanYangZhou
ComplexWarning in VJP when using complex matrix multiplication
bug
Something isn't working
#21188
opened May 11, 2024 by
mathisgerdes
Compilation cache does not work with custom partitioning
enhancement
New feature or request
#21159
opened May 10, 2024 by
jaro-sevcik
Strided indexing turns into gather
enhancement
New feature or request
performance
make things lean and fast
#21151
opened May 9, 2024 by
jewillco
[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 Something isn't working
eval_jaxpr
with 0.4.27
bug
#21116
opened May 7, 2024 by
patrick-kidger
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
Persistent compilation cache does not work
bug
Something isn't working
#21067
opened May 3, 2024 by
neel04
Previous Next
ProTip!
Type g i on any issue or pull request to go back to the issue listing page.