Skip to content

Issues: google/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,333 Open 3,880 Closed
1,333 Open 3,880 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

PyTreeDef difference not visible to humans, but assert fails bug Something isn't working
#21581 opened Jun 1, 2024 by johannahaffner
jax-metal: argmax fails with unsigned index type bug Something isn't working
#21577 opened Jun 1, 2024 by jonatanklosko
Involuntary full rematerialization for advanced data movements enhancement New feature or request
#21562 opened May 31, 2024 by chaserileyroberts
3
constants in Pallas kernel causes ValueError: safe_zip() argument 2 is shorter than argument 1 bug Something isn't working
#21557 opened May 31, 2024 by zhixuan-lin
jax-metal: 3-D convolution crashes Apple GPU (Metal) plugin bug Something isn't working
#21554 opened May 31, 2024 by jonatanklosko
@shuhand0
jax-metal: segmentation fault inside jax.lax.while_loop Apple GPU (Metal) plugin bug Something isn't working
#21552 opened May 31, 2024 by jonatanklosko
@shuhand0
1
jax-metal: gather fails with unsigned indices Apple GPU (Metal) plugin bug Something isn't working
#21547 opened May 31, 2024 by jonatanklosko
@shuhand0
jax raise AttributeError on import when running with python doctring strip optimization bug Something isn't working
#21542 opened May 31, 2024 by zazbone
jax.scipy.special.hyp1f1 unstable where scipy.special.hyp1f1 is not bug Something isn't working
#21503 opened May 30, 2024 by HansN87
1
softmax returns NaN when inputs are inf bug Something isn't working
#21495 opened May 29, 2024 by jakevdp
1
@jakevdp
2
jax/jaxlib not working bug Something isn't working
#21491 opened May 29, 2024 by RivasLab
@dfm
5
Cryptic error message when running pallas kernel on TPU bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#21488 opened May 29, 2024 by finbarrtimbers
3
Can´t get jax profiling to work bug Something isn't working
#21483 opened May 29, 2024 by MaHaWo
2
Slow initialisation on H100 bug Something isn't working
#21474 opened May 29, 2024 by ZacCranko
1
Tracking issue: inline docstrings documentation
#21461 opened May 28, 2024 by jakevdp
@jakevdp
2
example path bug Something isn't working
#21459 opened May 28, 2024 by sycamoreoak
1
D2H (gpu -> cpu) transfer via device_put is very slow bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#21438 opened May 27, 2024 by lengstrom
7
Low calculation performance compared to autograd elementwise_grad bug Something isn't working
#21436 opened May 26, 2024 by krysros
1
Unexpected scaling of sharding/pmap bug Something isn't working
#21411 opened May 24, 2024 by lockwo
1
unsafe_rbg fold in function is insensitive to the order data is folded in. bug Something isn't working
#21405 opened May 23, 2024 by justinjfu
1
jax-metal: top-k invalid behaviour with NaNs Apple GPU (Metal) plugin bug Something isn't working
#21397 opened May 23, 2024 by jonatanklosko
@kulinseth @shuhand0
Cryptic XLA error when using JAX 0.4.26 bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#21396 opened May 23, 2024 by itk22
9
jax-metal: inconsistent index clipping in jax.lax.dynamic_slice Apple GPU (Metal) plugin bug Something isn't working
#21393 opened May 23, 2024 by jonatanklosko
@shuhand0
jax-metal: inconsistent behaviour of overflowing update slice in jax.lax.dynamic_update_slice Apple GPU (Metal) plugin bug Something isn't working
#21392 opened May 23, 2024 by jonatanklosko
@shuhand0
jax-metal: count leading zeros not supported Apple GPU (Metal) plugin bug Something isn't working
#21389 opened May 23, 2024 by jonatanklosko
@kulinseth @shuhand0
2
ProTip! What’s not been updated in a month: updated:<2024-05-01.