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,470 Open 4,352 Closed
1,470 Open 4,352 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

Unexpected NaN gradient of jnp.abs at ±inf + 0j bug Something isn't working
#25681 opened Dec 24, 2024 by lfaucheux
Unexpected NaN when signing ±inf + 0j bug Something isn't working
#25679 opened Dec 23, 2024 by lfaucheux
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
jax.random.choice(replace=True) samples 0 probability index bug Something isn't working duplicate This issue or pull request already exists
#25498 opened Dec 16, 2024 by LemonATsu
1
jax.scipy.spatial.transform.Rotation.from_quat is missing scalar_first flag bug Something isn't working
#25491 opened Dec 14, 2024 by adam-hartshorne
@jakevdp
1
register_dataclass does not handle __init__ methods bug Something isn't working
#25486 opened Dec 13, 2024 by patrick-kidger
@jakevdp
3
Slow convolution, many memory warnings bug Something isn't working
#25461 opened Dec 13, 2024 by apasarkar
2
Segmentation fault in jaxlib 0.4.37 and Debian 11 ARM64 bug Something isn't working
#25436 opened Dec 12, 2024 by gongomgra
3
Replace np.array() with np.fromfile() to improve performance enhancement New feature or request
#25418 opened Dec 11, 2024 by perfNerdJK
Latency Hiding Scheduler not working with jax 0.4.35 bug Something isn't working
#25404 opened Dec 11, 2024 by lgiacomoni
1
jax.full allocates memory on the wrong device bug Something isn't working
#25396 opened Dec 11, 2024 by Chaosruler972
@yashk2810
2
Unable to parse backend config for custom call: Could not convert JSON string to proto: Expected : between key:value pair. bug Something isn't working
#25389 opened Dec 10, 2024 by adam-hartshorne
4
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.