Skip to content

Issues: jax-ml/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
670 Open 2,065 Closed
670 Open 2,065 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

Functions like jax.numpy.concatenate are not compatible with jax.sharding, resulting in improper functioning bug Something isn't working
#24576 opened Oct 29, 2024 by kYangLi
MaxText Getting Started fails bug Something isn't working
#24549 opened Oct 28, 2024 by learning-to-play
1
segmentation fault from a simple test on GCP vm bug Something isn't working
#24548 opened Oct 27, 2024 by jiyuuchc
2
return in finally swallows exceptions bug Something isn't working
#24541 opened Oct 25, 2024 by iritkatriel
1
[Pallas][Flash Attention] Cost analysis in Flash Attention kernel causing compilation OOM bug Something isn't working
#24539 opened Oct 25, 2024 by lsy323
INVALID_ARGUMENT when copying array in pinned_host memory with device_put(..., may_alias=False) bug Something isn't working
#24521 opened Oct 25, 2024 by ojw28
Get wrong structure when using jax2tf to convert nnx.module into tflite file bug Something isn't working
#24497 opened Oct 24, 2024 by noahzhy
1
Matmul operandlhs must have a shape divisible by (16, 1), but got: (8, 2048) bug Something isn't working
#24486 opened Oct 23, 2024 by vanbasten23
Attribute Error bug Something isn't working
#24478 opened Oct 23, 2024 by Opiyo-Inno
@dougalm
17
[Pallas TPU] Unable to create a boolean array inside kernel due to Mosaic relayout error bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24464 opened Oct 22, 2024 by ayaka14732
@ayaka14732
1
Full coverage for __jax_array__ protocol bug Something isn't working
#24460 opened Oct 22, 2024 by maxencefaldor
@mattjj
1
[ROCM] Sharding leads to HIP errors bug Something isn't working
#24458 opened Oct 22, 2024 by PhilipVinc
Inconsistency in using == operator on a jax array and a list bug Something isn't working
#24454 opened Oct 22, 2024 by gokul-uf
@jakevdp
1
shard_map - NotImplementedError: No replication rule for cond bug Something isn't working
#24418 opened Oct 21, 2024 by giovannicemin
2
Not implemented: Non-trivial layouts unsupported bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24415 opened Oct 20, 2024 by vanbasten23
4
jax.lax.approx_min_k() on GPU appears to be a brute force search bug Something isn't working
#24414 opened Oct 20, 2024 by notnot
JAX code is extremely slow on GPUs bug Something isn't working performance make things lean and fast
#24411 opened Oct 20, 2024 by AakashKumarNain
@mattjj
11
docs: pprof command is out of date bug Something isn't working
#24407 opened Oct 19, 2024 by garymm
1
INTERNAL: Mosaic failed to compile TPU kernel: unsupported shape cast bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24404 opened Oct 19, 2024 by vanbasten23
4
jax.distributed.initialize() crash bug Something isn't working
#24399 opened Oct 19, 2024 by demon2036
4
jax.jit changes the key order of returned dictionaries bug Something isn't working
#24398 opened Oct 18, 2024 by carlosgmartin
6
jax.jit slows down the code a lot on function with simple array operations and jnp.roll() bug Something isn't working XLA
#24373 opened Oct 17, 2024 by pmocz
4
[Pallas TPU] Mosaic legalisation error caused by select(b, 1, 0) being eliminated bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24367 opened Oct 17, 2024 by ayaka14732
@ayaka14732
ValueError when attempting to AOT lower or remat functions with jax_getattr. bug Something isn't working
#24314 opened Oct 15, 2024 by markblee
Matrix factorization on multiple GPUs causes memory overflow bug Something isn't working
#24309 opened Oct 15, 2024 by ChenAo-Phys
3
ProTip! What’s not been updated in a month: updated:<2024-09-29.