-
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
Custom partitioning error in fused_attention_stablehlo
bug
Something isn't working
#21584
opened Jun 2, 2024 by
kvablack
Guidelines on reducing compilation memory?
enhancement
New feature or request
#21583
opened Jun 2, 2024 by
OhadRubin
str(PyTreeDef) identical for two PyTreeDefs, but assert with allclose 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
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
Previous Next
ProTip!
no:milestone will show everything without a milestone.