-
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
Passed New feature or request
stacklevel=2
for the new tree_flatten
deprecation warning.
enhancement
#21833
opened Jun 12, 2024 by
patrick-kidger
jit-eval_shape-<callable PyTree with numpy __eq__ semantics>
crashes on JAX 0.4.29
bug
#21825
opened Jun 12, 2024 by
patrick-kidger
jax-metal: minimum/maximum/min/max/argmin/argmax invalid output with NaNs
Apple GPU (Metal) plugin
bug
Something isn't working
#21821
opened Jun 12, 2024 by
jonatanklosko
pmax bug on negative numbers on CPU
bug
Something isn't working
#21820
opened Jun 12, 2024 by
bondquant
[ROCM] Error: wheel file is invalid
AMD GPU
Issues pertaining to AMD GPUs (ROCM)
bug
Something isn't working
#21816
opened Jun 11, 2024 by
PhilipVinc
jax.debug.print printing empyt output
question
Questions for the JAX team
#21792
opened Jun 11, 2024 by
xudong2019
jax.numpy.linalg.inv returns spurious results when called with array of matrices
bug
Something isn't working
duplicate
This issue or pull request already exists
#21785
opened Jun 10, 2024 by
johannahaffner
jax.scipy.special.expi extremely slow when applied to a jnp.array
bug
Something isn't working
#21760
opened Jun 10, 2024 by
AntLrc
jax-metal: non-deterministic behavior of Something isn't working
jnp.take_along_axis
Apple GPU (Metal) plugin
bug
#21757
opened Jun 10, 2024 by
jkarwowski
TracerBoolConversionError when jitting jax.numpy.linalg.norm
bug
Something isn't working
#21730
opened Jun 7, 2024 by
mvilucchio
partial eval silently skips effects
question
Questions for the JAX team
#21713
opened Jun 6, 2024 by
Alan-Chen99
ndarray.at.set(mode="drop") gives incorrect value for the last element in the array
bug
Something isn't working
#21708
opened Jun 6, 2024 by
Gregory-Meyer
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
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
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
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: segmentation fault inside Something isn't working
jax.lax.while_loop
Apple GPU (Metal) plugin
bug
#21552
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
Previous Next
ProTip!
Updated in the last three days: updated:>2024-06-09.