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,299 Open 3,852 Closed
1,299 Open 3,852 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

Numerical differences between shardings in random algorithm bug Something isn't working
#21232 opened May 14, 2024 by shawnwang18
@froystig
1
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
2
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 on TPU 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
spsolve exits with error when inverting matrix sum bug Something isn't working
#21031 opened May 1, 2024 by buvoli
@jakevdp
4
jnp.fft.ifft imprecision for GPU bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#21002 opened Apr 30, 2024 by chih-kang-huang
Sharding is much slower than pmap for while loops of varying length while loops bug Something isn't working
#20968 opened Apr 27, 2024 by lockwo
4
Counterintuitive speed of einsums vs equivalent matmuls bug Something isn't working
#20952 opened Apr 26, 2024 by shehzaidi
5
Performance Issue Report: JAX Slower Than Autograd on GPU and CPU Setups bug Something isn't working type:performance
#20948 opened Apr 26, 2024 by alibastami
4
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.