-
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
pmap out_axes=None doesn't check if output is mapped/unmapped
bug
Something isn't working
P2 (eventual)
This ought to be addressed, but has no schedule at the moment. (Assignee optional)
#21671
opened Jun 5, 2024 by
mattjj
Inconsistent results with shard_map when switching PRNG Implementation from threefry2x32 to rbg
documentation
#21663
opened Jun 5, 2024 by
southfreebird
Using Something isn't working
needs info
More information is required to diagnose & prioritize the issue.
jax.config.update
within config context manager fails to set new value
bug
#21650
opened Jun 4, 2024 by
jakevdp
Wrong array values in Something isn't working
jax.debug.print
and the actual results when using lax.slice
in lax.scan
bug
#21637
opened Jun 4, 2024 by
mfkasim1
Wrong results on matmul's associative_scan when jitted within scan
bug
Something isn't working
#21628
opened Jun 4, 2024 by
mfkasim1
jaxlib.xla_extension.XlaRuntimeError: INTERNAL: ptxas exited with non-zero error code 11
bug
Something isn't working
#21621
opened Jun 4, 2024 by
PhilipVinc
support batched matrix multiplication in pallas
enhancement
New feature or request
pallas
Issues pertaining to Pallas (GPU or TPU)
#21618
opened Jun 4, 2024 by
zhixuan-lin
jax-metal: cond fails in compile in certain cases
Apple GPU (Metal) plugin
bug
Something isn't working
#21601
opened Jun 3, 2024 by
jonatanklosko
jax-metal: dynamic update slice fails with unsigned indices
Apple GPU (Metal) plugin
bug
Something isn't working
#21590
opened Jun 3, 2024 by
jonatanklosko
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
Apple GPU (Metal) plugin
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
pallas
Issues pertaining to Pallas (GPU or TPU)
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.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
Previous Next
ProTip!
Find all open issues with in progress development work with linked:pr.