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,386 Open 4,216 Closed
1,386 Open 4,216 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

FlaxLlamaForCausalLMModule hanging on jax-metal bug Something isn't working
#24221 opened Oct 9, 2024 by alexlatif
2
Computation hangs forever with jax > 0.4.30 bug Something isn't working
#24219 opened Oct 9, 2024 by mfschubert
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
10
Support for CUDNN 9.0? enhancement New feature or request
#24180 opened Oct 8, 2024 by jbkyang-nvi
3
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
[docs][FR] document how to build asan jaxlib documentation enhancement New feature or request
#24133 opened Oct 4, 2024 by skye
Problems when customizing event-driven opertors with pallas bug Something isn't working
#24117 opened Oct 4, 2024 by chaoming0625
@justinjfu
1
Offloading in grad(scan(remat(fn, policy=offload))) results in XlaRuntimeError bug Something isn't working
#24115 opened Oct 4, 2024 by hbq1
@yashk2810
1
[Regression] Gradient explodes after upgrading to JAX 0.4.33 from 0.4.30 bug Something isn't working
#24114 opened Oct 4, 2024 by qGentry
@dfm
3
scale_and_translate with output_size and scale returning zero values for last row and column bug Something isn't working
#24106 opened Oct 3, 2024 by barney-s
@dfm
1
Issues with vmap and GPU memory issues bug Something isn't working
#24099 opened Oct 3, 2024 by rruizdeaustri
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
4
Regression in JAX FP8 matmul fusion bug Something isn't working
#24051 opened Oct 1, 2024 by balancap
4
dot_product_attention has inconsistent dot types in backward pass bug Something isn't working
#24047 opened Oct 1, 2024 by sbodenstein
6
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
1
@ayaka14732
1
[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
3
ProTip! Updated in the last three days: updated:>2024-10-07.