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

1,476 Open 4,355 Closed
1,476 Open 4,355 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

If CUDA 12.1 is installed, pip-installed ptxas binary is not used and jax throws an error bug Something isn't working
#25718 opened Jan 3, 2025 by takkyu2
GPU pallas_call loses compiler params during second call when double jit-wrapped bug Something isn't working
#25714 opened Jan 3, 2025 by hanzhi713
JAX can create str-dtyped tracers under eval_shape with numpy 2 bug Something isn't working
#25707 opened Jan 2, 2025 by patrick-kidger
@jakevdp
1
Quickstart: using JIT-compiled functions, jax.vmap does not provide significant speedup over naive implementation question Questions for the JAX team
#25703 opened Dec 31, 2024 by zsamboki
@jakevdp
1
Get "invalid value (nan) encountered in jit" even when jit disabled globally bug Something isn't working
#25701 opened Dec 31, 2024 by SUSYUSTC
@emilyfertig
7
jax.experimental.jet.jet series argument: unintuitive ordering; at least documentation should explain it better bug Something isn't working
#25700 opened Dec 31, 2024 by tjltjl
@mattjj
Batched meshgrid or alternative enhancement New feature or request
#25696 opened Dec 30, 2024 by Qazalbash
1
debugging/make_color_iter() returns the same index when the number of colors is 2 and do not use all the possible colors bug Something isn't working
#25695 opened Dec 30, 2024 by jeromemassot
Differentiation rule for tridiagonal_solve enhancement New feature or request
#25693 opened Dec 29, 2024 by segasai
@dfm
2
Unexpected NaN gradient of jnp.abs at ±inf + 0j bug Something isn't working
#25681 opened Dec 24, 2024 by lfaucheux
@dfm
Unexpected NaN when signing ±inf + 0j bug Something isn't working
#25679 opened Dec 23, 2024 by lfaucheux
@jakevdp
2
Crash with jit of ordered io_callback under shard_map and then no shard_map on CPU bug Something isn't working
#25671 opened Dec 23, 2024 by jburnim
@yashk2810
1
⚠️ Nightly upstream-dev CI failed ⚠️ CI
#25664 opened Dec 23, 2024 by github-actions bot
@jakevdp
2
tree_util error handling throws error bug Something isn't working
#25659 opened Dec 22, 2024 by Sohl-Dickstein
@mattjj
1
Got "Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR" with TF 2.18 and jax with cuda_local bug Something isn't working
#25658 opened Dec 22, 2024 by davidshen84
Add mode='fan_geom_avg' to nn.initializers.variance_scaling enhancement New feature or request
#25649 opened Dec 20, 2024 by carlosgmartin
Improvements to debug_nans enhancement New feature or request
#25643 opened Dec 20, 2024 by emilyfertig
5 tasks
Add mask argument to lax.argmax enhancement New feature or request
#25623 opened Dec 20, 2024 by carlosgmartin
9
Add replace: bool argument to random.categorical to sample without replacement using Gumbel-top-k trick enhancement New feature or request
#25617 opened Dec 19, 2024 by carlosgmartin
@jakevdp
2
Surprising difference of output between NumPy's float32 and JAX's float32 bug Something isn't working
#25601 opened Dec 19, 2024 by jeertmans
8
error: 'string' in namespace 'std' does not name a type bug Something isn't working
#25598 opened Dec 19, 2024 by adamjstewart
1
Error when following the JAX tutorial on "Introduction to parallel programming" bug Something isn't working
#25541 opened Dec 17, 2024 by Stella-S-Yan
16
Make xla_bridge.is_gpu more extensible enhancement New feature or request
#25521 opened Dec 17, 2024 by PragmaTwice
Clarification re: supported data types in jax.linearize and jax.linear_transpose enhancement New feature or request
#25517 opened Dec 16, 2024 by johannahaffner
Public API for mesh axis index size enhancement New feature or request
#25515 opened Dec 16, 2024 by gabbard
2
ProTip! What’s not been updated in a month: updated:<2024-12-03.