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,374 Open 4,207 Closed
1,374 Open 4,207 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

jnp.polyfit with cov=True only returns one covariance matrix for multiple right hand sides bug Something isn't working
#24073 opened Oct 2, 2024 by scottstanie
@dfm
Edge case: Normal CDF incorrect when called with extremely small integer value if location is specified as integer array bug Something isn't working
#24059 opened Oct 1, 2024 by jleugeri
@jakevdp
1
Non-closure version of jax.vjp enhancement New feature or request
#24053 opened Oct 1, 2024 by luyug
@mattjj
3
Regression in JAX FP8 matmul fusion bug Something isn't working
#24051 opened Oct 1, 2024 by balancap
3
dot_product_attention has inconsistent dot types in backward pass bug Something isn't working
#24047 opened Oct 1, 2024 by sbodenstein
1
Results to not match reference bug Something isn't working
#24036 opened Sep 30, 2024 by truncs
1
[Pallas TPU] Core dump when comparing two boolean arrays bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24030 opened Sep 30, 2024 by ayaka14732
@ayaka14732
[Pallas TPU] Core dump when using jnp.remainder bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24027 opened Sep 30, 2024 by ayaka14732
@ayaka14732
1
[Pallas TPU] Wrong value when using input_output_aliases with multiple arrays bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24023 opened Sep 30, 2024 by ayaka14732
@ayaka14732
[DOC] Incorrect interval in jax.nn.log_softmax enhancement New feature or request
#23992 opened Sep 28, 2024 by Qazalbash
1
Mosaic failed to compile TPU kernel bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#23989 opened Sep 28, 2024 by jhkchan
3
[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
1
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
1
@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
ProTip! Follow long discussions with comments:>50.