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,334 Open 4,175 Closed
1,334 Open 4,175 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

Orthogonal Initializer raises gpusolverDnCreate(&handle) failed: cuSolver internal error bug Something isn't working
#23616 opened Sep 13, 2024 by brianorbrain
io_callback does not work with custom_vjp enhancement New feature or request
#23614 opened Sep 13, 2024 by GeophyAI
@jakevdp
5
make jax.debug.print work with non-jax types enhancement New feature or request
#23600 opened Sep 12, 2024 by dlwh
2 tasks done
@yashk2810
4
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
Wrong results on CPU since 0.4.32 bug Something isn't working
#23590 opened Sep 12, 2024 by dionhaefner
5
closure_convert doesn't work on VJPs bug Something isn't working
#23588 opened Sep 12, 2024 by hbq1
Sparse matrix multiplication and output nse values enhancement New feature or request
#23576 opened Sep 11, 2024 by joaospinto
2
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
Error compile: make_batch_pointers.cu.cc failed: undeclared inclusion(s) bug Something isn't working
#23573 opened Sep 11, 2024 by johnnynunez
@dfm
13
[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
2
Unable to build jaxlib with debug symbols for GPU bug Something isn't working
#23564 opened Sep 11, 2024 by juuso-oskari
1
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
Unused vmap GPU memory allocation causes RESOURCE_EXHAUSTED for versions >0.4.14 bug Something isn't working
#23548 opened Sep 10, 2024 by pwithams
2
All-gather performed in fp32 instead of bf16 despite explicit datatype conversion bug Something isn't working
#23543 opened Sep 10, 2024 by qGentry
Project Plan Github Like Pytorch for features and next releases? enhancement New feature or request
#23537 opened Sep 10, 2024 by johnnynunez
@mattjj
1
add support for QR decomposition gradient calculation for wide matrices (rows < columns) enhancement New feature or request
#23533 opened Sep 10, 2024 by rlucas7
2 tasks done
[ROCM] x64 mode crashes with "redzone_checker with block dimensions: 1024x1x1: hipError_t" AMD GPU Issues pertaining to AMD GPUs (ROCM) bug Something isn't working XLA
#23506 opened Sep 9, 2024 by PhilipVinc
@mrodden
5
vmap(SPMD axis)/shmap/(vmap with capture) pattern breaks batching bug Something isn't working
#23476 opened Sep 6, 2024 by jkr26
@mattjj
1
Operator jax.lax.gather consumes a lot of memory on CPU bug Something isn't working
#23457 opened Sep 5, 2024 by daskol
6
Multi-process GPU jobs fail on Slurm bug Something isn't working
#23452 opened Sep 5, 2024 by ChenAo-Phys
6
TPU operations don't appear if profile is too long bug Something isn't working
#23446 opened Sep 5, 2024 by jlin816
ValueError: Apply must be called with an RNG as the third argument, the required signature is: apply(params, state, rng, *a, **k). The object was of type <class 'jax._src.interpreters.partial_eval.DynamicJaxprTracer'>: Traced<ShapedArray(uint32[2])>with<DynamicJaxprTrace(level=1/0) bug Something isn't working
#23428 opened Sep 4, 2024 by eaven123
@jakevdp
5
Profile shows TPU operations before CPU instructions bug Something isn't working
#23426 opened Sep 4, 2024 by jlin816
ProTip! Follow long discussions with comments:>50.