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,342 Open 4,189 Closed
1,342 Open 4,189 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

⚠️ Nightly upstream-dev CI failed ⚠️ CI
#23759 opened Sep 19, 2024 by github-actions bot
@jakevdp
2
Numerical implementation of logarithmic integral function enhancement New feature or request P3 (no schedule) We have no plan to work on this and, if it is unassigned, we would be happy to review a PR
#23732 opened Sep 18, 2024 by Qazalbash
1
[jax 0.4.33] XlaRuntimeError: FAILED_PRECONDITION: Buffer Definition Event: Unsupported number of sorted inputs: 17 bug Something isn't working
#23727 opened Sep 18, 2024 by PhilipVinc
2
@penpornk
2
Typing improvement: Preserve the wrapped callable's signature in jax.jit enhancement New feature or request
#23719 opened Sep 18, 2024 by lebrice
2 tasks done
1
cuDNN attention segmentation fault bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#23701 opened Sep 17, 2024 by sbodenstein
4
OptimSteps not compatible with shard_map due to lax.cond bug Something isn't working
#23693 opened Sep 17, 2024 by Rocamonde
@mattjj
2
jaxlib 0.4.32, external CUDA, gcc: error: unrecognized command-line option '--cuda-path=external/cuda_nvcc' bug Something isn't working
#23689 opened Sep 17, 2024 by adamjstewart
2
jax.numpy.nonzero extremely slow on sharded array bug Something isn't working
#23675 opened Sep 16, 2024 by muchanem
@yashk2810
2
Document if differentiability of implementation is dependent on hardware enhancement New feature or request
#23650 opened Sep 16, 2024 by unalmis
5
all_gather in pmap fails with new-style random keys, works with old-style bug Something isn't working
#23647 opened Sep 15, 2024 by Lookatator
@justinjfu
When I compute gradients and perform updates using the same values in Torch and JAX, I find that after multiple iterations, the inference results differ significantly. bug Something isn't working
#23646 opened Sep 15, 2024 by CZXIANGOvO
3
when a tracer error happens in for_loop, should point to the user's body function better_errors Improve the error reporting
#23637 opened Sep 13, 2024 by mattjj
Corner-case sharding regression when replacing concrete mesh with abstract mesh bug Something isn't working
#23634 opened Sep 13, 2024 by jkr26
2
jax.lax.linalg.lu returns LU factorisation for singular matrix, jax.scipy.linalg.lu_solve silently returns spurious result in 32bit bug Something isn't working
#23626 opened Sep 13, 2024 by johannahaffner
@jakevdp
6
When calculating the loss, the input data does not contain NaN, but the output contains NaN bug Something isn't working
#23625 opened Sep 13, 2024 by CZXIANGOvO
3
pure_callback is broken with multiple vmap bug Something isn't working
#23624 opened Sep 13, 2024 by Joshuaalbert
7
Orthogonal Initializer raises gpusolverDnCreate(&handle) failed: cuSolver internal error bug Something isn't working
#23616 opened Sep 13, 2024 by brianorbrain
@dfm
3
Make jax.distributed timeouts configurable via jax.config enhancement New feature or request
#23599 opened Sep 12, 2024 by skye
Allow tuple inputs to scatter_dimension in jax.lax.psum_scatter enhancement New feature or request
#23594 opened Sep 12, 2024 by vvvm23
Sparse matrix multiplication and output nse values enhancement New feature or request
#23576 opened Sep 11, 2024 by joaospinto
7
Error compile: --bazel_options=--repo_env=LOCAL_CUDA_PATH="${CUDA_HOME}" issues with clang & gcc bug Something isn't working
#23575 opened Sep 11, 2024 by benkirk
10
[ROCM] Multi-device reduction causes segfault AMD GPU Issues pertaining to AMD GPUs (ROCM) bug Something isn't working
#23565 opened Sep 11, 2024 by PhilipVinc
@JehandadKhan
3
Unable to build jaxlib with debug symbols for GPU bug Something isn't working
#23564 opened Sep 11, 2024 by juuso-oskari
2
fp32 or even fp64 support for cudnn flashattention enhancement New feature or request
#23560 opened Sep 11, 2024 by wangleiphy
1
jax.debug.breakpoint gives UnexpectedTracerError when used with jax.lax.cond bug Something isn't working
#23555 opened Sep 10, 2024 by billmark
3
ProTip! Adding no:label will show everything without a label.