Skip to content

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.

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
61 Open 265 Closed
61 Open 265 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

NotImplementedError: outfeed rewrite custom_linear_solve enhancement New feature or request
#10112 opened Mar 31, 2022 by denizokt
@sharadmv
error handling for CustomCall implementation of host_callback enhancement New feature or request
#11424 opened Jul 8, 2022 by YouJiacheng
@sharadmv
BUG: jax_debug_nans can cause hangs when used w/ multiprocess pmap bug Something isn't working
#11987 opened Aug 18, 2022 by sharadmv
@sharadmv
Scan as for
#12363 opened Sep 14, 2022 by sharadmv • Draft
@sharadmv
JEP: JAX Foreign Function Interface JEP JAX enhancement proposal
#12535 opened Sep 27, 2022 by sharadmv
@zhangqiaorjc @sharadmv @cky9301
Upgrade the state primitive traceables to handle more forms of indexing
#12708 opened Oct 8, 2022 by sharadmv Loading…
@sharadmv
1
Fix wrong cond-state discharge rule. pull ready Ready for copybara import and testing
#13785 opened Dec 24, 2022 by patrick-kidger Loading…
@sharadmv
1
Added xla_call and custom_jvp_call discharge rules pull ready Ready for copybara import and testing
#13837 opened Dec 31, 2022 by patrick-kidger Loading…
@sharadmv
jax.debug.breakpoint(num_frames=1) crashes in colab bug Something isn't working
#16187 opened May 30, 2023 by patrick-kidger
@sharadmv
Add a check to the jaxpr type checker that verifies that the same Ref isn't passed into an equation multiple times enhancement New feature or request
#16456 opened Jun 16, 2023 by sharadmv
@sharadmv @mattjj
Change scan partial_eval to use partial_eval_jaxpr_dynamic
#17357 opened Aug 29, 2023 by sharadmv Loading…
@sharadmv
Better error messages for shape and dtype mismatch.
#17546 opened Sep 11, 2023 by mehdiataei Loading…
@sharadmv
Allow import or other statements in jdb enhancement New feature or request
#18059 opened Oct 11, 2023 by jjyyxx
2 tasks done
@sharadmv
Wrong behaviour of pallas.atomic_cas bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs pallas Issues pertaining to Pallas (GPU or TPU)
#18909 opened Dec 10, 2023 by southfreebird
@sharadmv @apaszke
Pallas: jnp.indices fails to lower to mosaic bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#19291 opened Jan 10, 2024 by j2kun
@sharadmv
pallas hello world not implemented for int8 bug Something isn't working pallas Issues pertaining to Pallas (GPU or TPU)
#22204 opened Jun 30, 2024 by vlad17
@sharadmv
[Pallas/TPU] Correctly populate new_avals when we are using sem indexers in dma_start discharge rule
#23135 opened Aug 20, 2024 by copybara-service bot Loading…
@sharadmv
Add memory space annotation to ShapedArray
#23633 opened Sep 13, 2024 by copybara-service bot Loading…
@sharadmv
[Pallas TPU] Add support for passing in and returning semaphores
#23681 opened Sep 17, 2024 by copybara-service bot Loading…
@sharadmv
Adding host_callbacks.id_tap reorders JVP evaluation bug Something isn't working
#3198 opened May 24, 2020 by shoyer
@sharadmv @gnecula
1
Fix Cmd.use_rawinput type in debuggers
#12231 opened Sep 6, 2022 by sobolevn Loading…
@sharadmv
2
JEP: Unified Error Handling API JEP JAX enhancement proposal
#12537 opened Sep 27, 2022 by sharadmv
@sharadmv @mattjj @LenaMartens
1
support different reverse settings for different args in scan enhancement New feature or request
#12584 opened Sep 29, 2022 by GallagherCommaJack
@sharadmv
1
checkify does not seem to check "constant" computations bug Something isn't working
#13318 opened Nov 18, 2022 by billmark
@sharadmv
1
Proposal to add lax.gather_scatter_add to support embedding bags and other features enhancement New feature or request
#13354 opened Nov 22, 2022 by EthanSteinberg
@hawkinsp @sharadmv
1
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.