-
Notifications
You must be signed in to change notification settings - Fork 2.6k
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.
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
Author
Label
Projects
Milestones
Assignee
Sort
Issues list
Involuntary full rematerialization for advanced data movements
enhancement
New feature or request
#21562
opened May 31, 2024 by
chaserileyroberts
constants in Pallas kernel causes Something isn't working
ValueError: safe_zip() argument 2 is shorter than argument 1
bug
#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
jax-metal: segmentation fault inside Something isn't working
jax.lax.while_loop
Apple GPU (Metal) plugin
bug
#21552
opened May 31, 2024 by
jonatanklosko
jax-metal: gather fails with unsigned indices
Apple GPU (Metal) plugin
bug
Something isn't working
#21547
opened May 31, 2024 by
jonatanklosko
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
softmax
returns NaN
when inputs are inf
bug
#21495
opened May 29, 2024 by
jakevdp
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
D2H (gpu -> cpu) transfer via Something isn't working
NVIDIA GPU
Issues specific to NVIDIA GPUs
device_put
is very slow
bug
#21438
opened May 27, 2024 by
lengstrom
Low calculation performance compared to autograd elementwise_grad
bug
Something isn't working
#21436
opened May 26, 2024 by
krysros
Unexpected scaling of sharding/pmap
bug
Something isn't working
#21411
opened May 24, 2024 by
lockwo
unsafe_rbg
fold in function is insensitive to the order data is folded in.
bug
#21405
opened May 23, 2024 by
justinjfu
jax-metal: top-k invalid behaviour with NaNs
Apple GPU (Metal) plugin
bug
Something isn't working
#21397
opened May 23, 2024 by
jonatanklosko
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
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
jax-metal: inconsistent behaviour of overflowing update slice in Something isn't working
jax.lax.dynamic_update_slice
Apple GPU (Metal) plugin
bug
#21392
opened May 23, 2024 by
jonatanklosko
jax-metal: count leading zeros not supported
Apple GPU (Metal) plugin
bug
Something isn't working
#21389
opened May 23, 2024 by
jonatanklosko
jax-metal: reduce window not supported
Apple GPU (Metal) plugin
bug
Something isn't working
#21387
opened May 23, 2024 by
jonatanklosko
jax-metal: reduce with multiple operands failed to legalize
Apple GPU (Metal) plugin
bug
Something isn't working
#21384
opened May 23, 2024 by
jonatanklosko
Previous Next
ProTip!
Type g p on any issue or pull request to go back to the pull request listing page.