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

1,412 Open 4,239 Closed
1,412 Open 4,239 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

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
isAllowMockLocationsOn
#24494 opened Oct 24, 2024 by doni108108doni
Pallas Attention Kernel Wrong results when length of query != length of KV bug Something isn't working
#24491 opened Oct 23, 2024 by zinccat
1
2
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
6
[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
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
pmap of pjit enhancement New feature or request
#24435 opened Oct 21, 2024 by luyug
4
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
#24411 opened Oct 20, 2024 by AakashKumarNain
docs: pprof command is out of date bug Something isn't working
#24407 opened Oct 19, 2024 by garymm
1
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
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
1
jax.jit changes the key order of returned dictionaries bug Something isn't working
#24398 opened Oct 18, 2024 by carlosgmartin
5
Allowing HLO constants inside custom partition enhancement New feature or request
#24390 opened Oct 18, 2024 by ASKabalan
2 tasks done
3
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
CPU profiling (not tracing) enhancement New feature or request
#24349 opened Oct 16, 2024 by joaospinto
4
Pallas autotuning enhancement New feature or request
#24340 opened Oct 16, 2024 by lengstrom
1
ValueError when attempting to AOT lower or remat functions with jax_getattr. bug Something isn't working
#24314 opened Oct 15, 2024 by markblee
ProTip! What’s not been updated in a month: updated:<2024-09-24.