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

1,191 Open 3,219 Closed
1,191 Open 3,219 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
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

XLA_FLAGS is only parsed once under JAX, which lead to side effects in unit test bug Something isn't working
#17188 opened Aug 18, 2023 by qlzh727
vmap of pure_callback(..., vectorized=True) fails on broadcasts bug Something isn't working
#17187 opened Aug 18, 2023 by Gattocrucco
@sharadmv
5
Support FP8 GEMM kernels which are already support in OpenXLA. enhancement New feature or request type:support
#17176 opened Aug 18, 2023 by MoFHeka
2 tasks done
vmap with indexing seems to be broken in jax-metal Apple GPU (Metal) plugin bug Something isn't working
#17175 opened Aug 18, 2023 by saran-t
@kulinseth @shuhand0
1
Jaxlib fails to build from source on Windows bug Something isn't working build contributions welcome The JAX team has not prioritized work on this. Community contributions are welcome. Windows Issues related to JAX on Microsoft Windows
#17174 opened Aug 18, 2023 by Sha-x2-nk
5
jnp.tan and jnp.arccos give incorrect values for certain complex numbers bug Something isn't working
#17172 opened Aug 18, 2023 by jshepherd01
1
type <class 'jaxlib.xla_extension.ArrayImpl'>is non-hashable when I use a class member function, while there is no problem when I use a general function. question Questions for the JAX team
#17158 opened Aug 17, 2023 by zoeStartover
@jakevdp
4
disable_jit array promotion issue bug Something isn't working
#17156 opened Aug 17, 2023 by cgiovanetti
2
ENH: Reflect changes from numpy namespace refactor Part 3 enhancement New feature or request
#17155 opened Aug 17, 2023 by mtsokol
@jakevdp
9
sparse_test_gpu start to fail when JAX is built w/ cuda-12.2.1 bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#17118 opened Aug 14, 2023 by Artem-B
1
@jakevdp
1
where related gradient bug in jax.nn.softmax and jax.nn.log_softmax (unprotected use of jnp.exp?) bug Something isn't working
#17117 opened Aug 14, 2023 by hrbigelow
jax and xarray integration for automatic differentiation? enhancement New feature or request
#17107 opened Aug 14, 2023 by tylerflex
2 tasks done
11
Compilation from source with Metal support fails Apple GPU (Metal) plugin bug Something isn't working
#17095 opened Aug 14, 2023 by mstoelzle
@kulinseth @shuhand0
1
build jaxlib from source bug Something isn't working build needs info More information is required to diagnose & prioritize the issue.
#17093 opened Aug 14, 2023 by never-to-never
1
Allow shard_map to support kwargs. enhancement New feature or request
#17077 opened Aug 11, 2023 by salayatana66
Compilation Error trying to Build jaxlib Apple GPU (Metal) plugin bug Something isn't working type:support
#17045 opened Aug 9, 2023 by dbl001
@kulinseth @shuhand0
5
Vectorizing over None argument returns NaNs bug Something isn't working
#17020 opened Aug 8, 2023 by AdrienCorenflos
@jakevdp
6
Nesting vmap within pmap using jax.pure_callback segfaults bug Something isn't working
#17001 opened Aug 7, 2023 by Justin-Tan
@sharadmv
jaxlib.xla_extension.XlaRuntimeError: UNIMPLEMENTED: batch in most minor dimension bug Something isn't working needs info More information is required to diagnose & prioritize the issue. NVIDIA GPU Issues specific to NVIDIA GPUs XLA
#16991 opened Aug 6, 2023 by alberthli
5
⚠️ Nightly upstream-dev CI failed ⚠️ CI
#16989 opened Aug 5, 2023 by github-actions bot
@jakevdp
1
arithmetic operations not implemented for BCSR enhancement New feature or request P3 (no schedule) We have no plan to work on this and, if it is unassigned, we would be happy to review a PR
#16922 opened Aug 2, 2023 by jiachenguoNU
@jakevdp
7
Simple optimization problem running faster in Tensor Flow compared to JAX with Optax bug Something isn't working
#16916 opened Aug 1, 2023 by alexandermm
1
Sharded FFT with JIT gives incorrect results on GPU since 0.4.9 bug Something isn't working
#16909 opened Jul 31, 2023 by Findus23
3
Fast path for device_put enhancement New feature or request
#16905 opened Jul 31, 2023 by jjyyxx
2 tasks done
@hawkinsp
3
Possible BUG: Different behavior of 64-bit integers when used inside Sharding ops bug Something isn't working needs info More information is required to diagnose & prioritize the issue.
#16903 opened Jul 31, 2023 by yashjha123
@yashk2810
2
ProTip! Add no:assignee to see everything that’s not assigned.