Skip to content

Issues: jax-ml/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

Clear current search query, filters, and sorts
16 Open 28 Closed
16 Open 28 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
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

[Pallas TPU] Unable to create a boolean array inside kernel due to Mosaic relayout error bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24464 opened Oct 22, 2024 by ayaka14732
@ayaka14732
1
[Pallas TPU] Mosaic legalisation error caused by select(b, 1, 0) being eliminated bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24367 opened Oct 17, 2024 by ayaka14732
@ayaka14732
[Pallas TPU] Error when negating a boolean value bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24243 opened Oct 11, 2024 by ayaka14732
@ayaka14732
3
[Pallas TPU] Wrong value when using input_output_aliases with multiple arrays bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#24023 opened Sep 30, 2024 by ayaka14732
@ayaka14732
1
jnp.pow() triggers dtype promotion when it should not bug Something isn't working
#23337 opened Aug 30, 2024 by ayaka14732
3
[Pallas] When mixing basic indexing and integer array indexing, the axis corresponding to integer array indexing is unnecessarily moved to the front bug Something isn't working
#22783 opened Jul 31, 2024 by ayaka14732
2
@ayaka14732
1
jaxlib or libtpu not detected on TPU Pod bug Something isn't working
#22070 opened Jun 24, 2024 by ayaka14732
@skye
9
Adding tree_util.stack_leaves() and tree_util.unstack_leaves() enhancement New feature or request
#20934 opened Apr 25, 2024 by ayaka14732
5
[Documentation] Move the Public API: jax package section up in the documentation enhancement New feature or request
#20535 opened Apr 2, 2024 by ayaka14732
1
@ayaka14732
7
jax.ensure_compile_time_eval() unexpectedly not failed when the computation is dependent on the argument enhancement New feature or request
#16515 opened Jun 22, 2023 by ayaka14732
1
Pickling a JAX array does not preserve its original device bug Something isn't working
#12253 opened Sep 7, 2022 by ayaka14732
@ayaka14732
13
jax.profiler does not fully work on TPU bug Something isn't working
#11409 opened Jul 8, 2022 by ayaka14732
2
vmap + array-indexing is very slow on TPU enhancement New feature or request performance make things lean and fast TPU Issues related to compilation for or performance on Google TPUs XLA
#10378 opened Apr 20, 2022 by ayaka14732
2
XLA: log-indexing should be optimised to indexing-log enhancement New feature or request XLA
#10319 opened Apr 16, 2022 by ayaka14732
3
'libtpu.so already in use' but actually not used bug Something isn't working
#10192 opened Apr 8, 2022 by ayaka14732
@skye
11
RuntimeError: INTERNAL: Core halted unexpectedly: No error message available as no compiler metadata was provided. bug Something isn't working
#9642 opened Feb 19, 2022 by ayaka14732
@skye
12
ProTip! What’s not been updated in a month: updated:<2024-09-29.