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
10 Open 36 Closed
10 Open 36 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] Fix integer array indexing pallas Issues pertaining to Pallas (GPU or TPU)
#23406 opened Sep 3, 2024 by ayaka14732 Loading…
1
@ayaka14732
1
Fix typing in 2 pallas tests
#23343 opened Aug 30, 2024 by copybara-service bot Loading…
@ayaka14732
[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
1
@ayaka14732
1
FileNotFoundError when using cache in parallel bug Something isn't working
#22718 opened Jul 29, 2024 by cool-RR
@ayaka14732
2
TPUVM program hangs when using jax_compilation_cache_dir with wrong gcs directory bug Something isn't working
#22364 opened Jul 10, 2024 by entrpn
@ayaka14732
9
[Pallas TPU] Add lowering for lax.erf_inv 64 bit pallas Issues pertaining to Pallas (GPU or TPU) pull ready Ready for copybara import and testing
#22310 opened Jul 8, 2024 by ayaka14732 Loading…
@ayaka14732
4
Adding multiprocessing tests for LRUCache
#21893 opened Jun 14, 2024 by ayaka14732 • Draft
@ayaka14732
3
Making JAX API Reference easier to find on documentation sidebar pull ready Ready for copybara import and testing
#20549 opened Apr 2, 2024 by ayaka14732 Loading…
1
@ayaka14732
1
[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
Pickling a JAX array does not preserve its original device bug Something isn't working
#12253 opened Sep 7, 2022 by ayaka14732
@ayaka14732
13
ProTip! Add no:assignee to see everything that’s not assigned.