-
Notifications
You must be signed in to change notification settings - Fork 2.8k
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.
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
Author
Label
Projects
Milestones
Assignee
Sort
Issues list
Pallas Attention Kernel Wrong results when length of query != length of KV
bug
Something isn't working
#24491
opened Oct 23, 2024 by
zinccat
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
[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
Full coverage for Something isn't working
__jax_array__
protocol
bug
#24460
opened Oct 22, 2024 by
maxencefaldor
[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
shard_map - NotImplementedError: No replication rule for cond
bug
Something isn't working
#24418
opened Oct 21, 2024 by
giovannicemin
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
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
#24411
opened Oct 20, 2024 by
AakashKumarNain
vmap gives misleading error message about inconsistent sizes when passing arg by keyword
bug
Something isn't working
#24406
opened Oct 19, 2024 by
garymm
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
jax.distributed.initialize() crash
bug
Something isn't working
#24399
opened Oct 19, 2024 by
demon2036
jax.jit changes the key order of returned dictionaries
bug
Something isn't working
#24398
opened Oct 18, 2024 by
carlosgmartin
Allowing HLO constants inside custom partition
enhancement
New feature or request
#24390
opened Oct 18, 2024 by
ASKabalan
2 tasks done
jax.jit
slows down the code a lot on function with simple array operations and jnp.roll()
bug
#24373
opened Oct 17, 2024 by
pmocz
[Pallas TPU] Mosaic legalisation error caused by Something isn't working
pallas
Issues pertaining to Pallas (GPU or TPU)
select(b, 1, 0)
being eliminated
bug
#24367
opened Oct 17, 2024 by
ayaka14732
CPU profiling (not tracing)
enhancement
New feature or request
#24349
opened Oct 16, 2024 by
joaospinto
ValueError
when attempting to AOT lower or remat functions with jax_getattr
.
bug
#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
Previous Next
ProTip!
Exclude everything labeled
bug
with -label:bug.