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,415 Open 4,247 Closed
1,415 Open 4,247 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

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
Allowing HLO constants inside custom partition enhancement New feature or request
#24390 opened Oct 18, 2024 by ASKabalan
2 tasks done
5
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
5
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
Matrix factorization on multiple GPUs causes memory overflow bug Something isn't working
#24309 opened Oct 15, 2024 by ChenAo-Phys
3
custom_root with integer aux output broken in 0.4.34 bug Something isn't working
#24295 opened Oct 14, 2024 by f0uriest
1
support for concat and dynamic slice in pallas/triton enhancement New feature or request pallas Issues pertaining to Pallas (GPU or TPU)
#24292 opened Oct 14, 2024 by juntang-zhuang
@justinjfu
1
argsort incorrectly handles very small floating-point numbers and -0.0 compared to PyTorch bug Something isn't working
#24280 opened Oct 14, 2024 by LilyDong0127
@hawkinsp
9
Arcsinh Error Report bug Something isn't working
#24276 opened Oct 14, 2024 by LilyDong0127
@jakevdp
5
lax.cond is much slower than calling true_fn itself when cond is always true. enhancement New feature or request
#24259 opened Oct 11, 2024 by hanzhi713
Pure callback with jax types hangs with jax > 0.4.31 bug Something isn't working
#24255 opened Oct 11, 2024 by mfschubert
6
[Pallas TPU] Error when negating a boolean value bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24243 opened Oct 11, 2024 by ayaka14732
@ayaka14732
3
FlaxLlamaForCausalLMModule hanging on jax-metal bug Something isn't working
#24221 opened Oct 9, 2024 by alexlatif
2
Support new dot Precision API in Pallas enhancement New feature or request
#24209 opened Oct 9, 2024 by sbodenstein
Inadequate memory consumption when using HSDP without gradient accumulation bug Something isn't working
#24208 opened Oct 9, 2024 by qGentry
ValueError when re-compiling function with a multi-dimensional array as a static field question Questions for the JAX team
#24204 opened Oct 9, 2024 by bheijden
@jakevdp
12
jax-metal slowdown on m3 max when external monitor is plugged in Apple GPU (Metal) plugin bug Something isn't working
#24163 opened Oct 7, 2024 by bjeurissen
@shuhand0
1
Unimplemented primitive in Pallas GPU lowering: dynamic_slice bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24145 opened Oct 6, 2024 by chaoming0625
3
Revisit the symbolic shape limitation on threefry_2x32 in terms of jax.export enhancement New feature or request
#24144 opened Oct 5, 2024 by jjyyxx
2 tasks done
Can not export function with input/output of typed keys bug Something isn't working
#24143 opened Oct 5, 2024 by jjyyxx
@gnecula
5
Errors when cusotmizing matrix-vector multiplication using pallas bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24142 opened Oct 5, 2024 by chaoming0625
ProTip! Exclude everything labeled bug with -label:bug.