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

Clear current search query, filters, and sorts
611 Open 2,008 Closed
611 Open 2,008 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
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
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
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
[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
_isolve tacitly checks for symmetric linear operator wrongly bug Something isn't working
#23403 opened Sep 3, 2024 by Joshuaalbert
4
error: failed to legalize operation 'mhlo.erf' with jax-metal bug Something isn't working
#23384 opened Sep 2, 2024 by a-gn
@kulinseth @shuhand0
Segmentation Fault on 8x8 or larger matrix multiplication with GPU bug Something isn't working
#23355 opened Aug 30, 2024 by treyra
1
jax.nn.dot_product_attention does not respect key_value_seq_lengths bug Something isn't working
#23349 opened Aug 30, 2024 by danjenson
5
jnp.pow() triggers dtype promotion when it should not bug Something isn't working
#23337 opened Aug 30, 2024 by ayaka14732
3
Full zero arrays allocated for "consumed" gradients of stop_gradient parameters bug Something isn't working
#23316 opened Aug 29, 2024 by colehaus
2
Cannot pass constant to lax.fori_loop body inside pallas kernel bug Something isn't working
#23301 opened Aug 28, 2024 by lengstrom
@justinjfu
1
jax.numpy.tan call adds chlo.tan call to exported StableHLO; stablehlo.tan should be preferred bug Something isn't working
#23259 opened Aug 26, 2024 by joaospinto
1
4
ProTip! Find all open issues with in progress development work with linked:pr.