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,370 Open 4,201 Closed
1,370 Open 4,201 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

[DOC] Incorrect interval in jax.nn.log_softmax enhancement New feature or request
#23992 opened Sep 28, 2024 by Qazalbash
Mosaic failed to compile TPU kernel bug Something isn't working
#23989 opened Sep 28, 2024 by jhkchan
2
[Pallas TPU] Unrelated 'maximum recursion depth exceeded' error when using .astype(jnp.int64) bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#23988 opened Sep 28, 2024 by ayaka14732
@ayaka14732
[Pallas TPU] jnp.uint32 comparisons give the wrong result with large numbers bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#23972 opened Sep 27, 2024 by ayaka14732
@ayaka14732
Ahead-of-time compilation on GPU fails without a GPU device bug Something isn't working
#23971 opened Sep 27, 2024 by jaro-sevcik
1
jitted function gives incorrect result in jnp.concatenate bug Something isn't working
#23958 opened Sep 26, 2024 by phdum-a
visualize_array_sharding/inspect_array_sharding doesn't work with shard_map bug Something isn't working
#23936 opened Sep 26, 2024 by Findus23
@mattjj
Add .at[].subtract enhancement New feature or request
#23933 opened Sep 26, 2024 by carlosgmartin
@superbobry
2
Segmentation Fault in jax-metal with variable indexing and partial slice in scan Apple GPU (Metal) plugin bug Something isn't working
#23931 opened Sep 25, 2024 by bsarkar321
[pallas] Add support for the new algorithm option to dot enhancement New feature or request pallas Issues pertaining to Pallas (GPU or TPU)
#23926 opened Sep 25, 2024 by sbodenstein
Add a fused multiply-add function enhancement New feature or request
#23919 opened Sep 25, 2024 by carlosgmartin
7
stop_gradient cannot safeguard io_callback against jax.grad transformation enhancement New feature or request
#23911 opened Sep 25, 2024 by ysngshn
Add override_type_lowerings to LoweringParameters enhancement New feature or request
#23904 opened Sep 25, 2024 by erick-xanadu
jax-metal: GPU crash(?) with large input Apple GPU (Metal) plugin bug Something isn't working
#23902 opened Sep 25, 2024 by matthewlai
@shuhand0
1
lax.cond is broken within compute_on('device_host') bug Something isn't working
#23887 opened Sep 24, 2024 by hanzhi713
@yashk2810
Jax[cuda] always allocates memory bug Something isn't working
#23882 opened Sep 24, 2024 by GJBoth
2
Checkify doesn't work with remat. bug Something isn't working
#23867 opened Sep 24, 2024 by giorgio-arena
@froystig
segfault with map and pure_callback bug Something isn't working
#23854 opened Sep 23, 2024 by Michael-T-McCann
Conversion from and to rotation vectors fails for identity rotations. bug Something isn't working
#23839 opened Sep 23, 2024 by jlindermeir
XlaRuntimeError: INTERNAL: unsupported operands type with float64 precision after 0.4.33 upgrade bug Something isn't working
#23835 opened Sep 23, 2024 by HGangloff
4
Unable to load any libcudnn_engines_precompiled.so using jax.nn.dot_product_attention bug Something isn't working
#23833 opened Sep 23, 2024 by zdcao121
3
BUG: jnp.fft.ifft: inaccuracy on CUDA for 64-bit array bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#23827 opened Sep 22, 2024 by lucascolley
@dfm
10
Implementation of PowerLaw distribution enhancement New feature or request
#23826 opened Sep 22, 2024 by Qazalbash
@jakevdp
paged_attention results in a core dumped on TPU-v3 bug Something isn't working
#23825 opened Sep 22, 2024 by OhadRubin
@justinjfu
1
JRT_Layouts_Extension is not found when PjRtCApiBuffer::layout is called when running basic example Apple GPU (Metal) plugin bug Something isn't working
#23820 opened Sep 21, 2024 by joaogui1
@shuhand0
5
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.