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,147 Open 3,078 Closed
1,147 Open 3,078 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

Stax has a bug, Dropout cannot convert from training mode to test mode bug Something isn't working
#16218 opened Jun 1, 2023 by fmscole
Many threads despite setting task affinities bug Something isn't working
#16215 opened Jun 1, 2023 by cooijmanstim
Bug with masking in jax.nn.log_softmax bug Something isn't working
#16208 opened Jun 1, 2023 by SanoVX
1
@jakevdp
1
Subnormal jax.numpy arrays equal to 0 bug Something isn't working
#16200 opened May 31, 2023 by honno
@hawkinsp
1
Calling a pjitted function repeatedly with the same inputs errors in multiprocessing execution bug Something isn't working
#16198 opened May 31, 2023 by terrykong
Sharding "devices" arg unexpected behavior bug Something isn't working
#16193 opened May 30, 2023 by wbrenton
@yashk2810
jax.debug.breakpoint(num_frames=1) crashes in colab bug Something isn't working
#16187 opened May 30, 2023 by patrick-kidger
@sharadmv
jax.debug.breakpoint crashes with numpy arrays of objects. bug Something isn't working
#16186 opened May 30, 2023 by patrick-kidger
@sharadmv
Is pjit version functionally identical to pmap? bug Something isn't working
#16184 opened May 30, 2023 by Young768
@yashk2810
2
When the inferred step is integer-valued, have linspace() return integer-valued results enhancement New feature or request
#16183 opened May 30, 2023 by honno
@jakevdp
1
jax.numpy.interp has poor performance on TPUs contributions welcome The JAX team has not prioritized work on this. Community contributions are welcome. enhancement New feature or request jax.numpy about JAX's NumPy implementation type:performance
#16182 opened May 30, 2023 by YunfanZhang42
6
Multi-machine multi-card support bug Something isn't working
#16172 opened May 30, 2023 by caixiiaoyang
jax.vmap fails for custom PyTree bug Something isn't working
#16170 opened May 29, 2023 by francois-rozet
1
@mattjj
4
Bad return values from jax.random.poisson when lam is jnp.inf bug Something isn't working
#16164 opened May 27, 2023 by JiaYaobo
@jakevdp
1
[shard-map] Check failed using unsafe_rbg_key bug Something isn't working
#16137 opened May 25, 2023 by mattjj
@froystig
A case where jnp.sinc + jit make a function unstable and non-deterministic bug Something isn't working
#16129 opened May 25, 2023 by astanziola
2
consider duck typing for jnp.ones_like, other similar functions enhancement New feature or request
#16122 opened May 24, 2023 by GallagherCommaJack
4
FAILED_PRECONDITION: DNN library initialization failed. when importing tensorflow before jax, with set_visible_devices bug Something isn't working
#16107 opened May 24, 2023 by jakiw
1
Very slow compile for tree-map over large number of leaves bug Something isn't working
#16084 opened May 22, 2023 by njwfish
5
JAX cannot create cudnn handle (CUDNN_STATUS_INTERNAL_ERROR) bug Something isn't working
#16078 opened May 20, 2023 by limesqueezy
@jakevdp
3
Please consider implementing differentiation for the Hessian of gamma variates 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
#16076 opened May 20, 2023 by NeilGirdhar
3
Support static arguments for any function transformation enhancement New feature or request
#16071 opened May 19, 2023 by danijar
2
Increase minimum CUDNN version to 8.9.1, and track CUDNN more closely in the future. enhancement New feature or request NVIDIA GPU Issues specific to NVIDIA GPUs
#16059 opened May 18, 2023 by hawkinsp
@yashk2810
2
einsum and einsum_path should calculate optimal paths when inputs have different vmap/batch shapes enhancement New feature or request
#16056 opened May 18, 2023 by ebrevdo
1
Implement to_numpy method to speed up matplotlib with JAX arrays enhancement New feature or request
#16054 opened May 18, 2023 by patel-zeel
2 tasks done
8
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.