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

Clear current search query, filters, and sorts
512 Open 744 Closed
512 Open 744 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

Allowing HLO constants inside custom partition enhancement New feature or request
#24390 opened Oct 18, 2024 by ASKabalan
2 tasks done
2
CPU profiling (not tracing) enhancement New feature or request
#24349 opened Oct 16, 2024 by joaospinto
4
Pallas autotuning enhancement New feature or request
#24340 opened Oct 16, 2024 by lengstrom
1
support for concat and dynamic slice in pallas/triton enhancement New feature or request pallas Issues pertaining to Pallas (GPU or TPU)
#24292 opened Oct 14, 2024 by juntang-zhuang
@justinjfu
1
lax.cond is much slower than calling true_fn itself when cond is always true. enhancement New feature or request
#24259 opened Oct 11, 2024 by hanzhi713
Support new dot Precision API in Pallas enhancement New feature or request
#24209 opened Oct 9, 2024 by sbodenstein
Revisit the symbolic shape limitation on threefry_2x32 in terms of jax.export enhancement New feature or request
#24144 opened Oct 5, 2024 by jjyyxx
2 tasks done
[docs][FR] document how to build asan jaxlib documentation enhancement New feature or request
#24133 opened Oct 4, 2024 by skye
Non-closure version of jax.vjp enhancement New feature or request
#24053 opened Oct 1, 2024 by luyug
@mattjj
4
[DOC] Incorrect interval in jax.nn.log_softmax enhancement New feature or request
#23992 opened Sep 28, 2024 by Qazalbash
1
[pallas] Add support for the new algorithm option to dot enhancement New feature or request pallas Issues pertaining to Pallas (GPU or TPU)
#23926 opened Sep 25, 2024 by sbodenstein
1
Add a fused multiply-add function enhancement New feature or request
#23919 opened Sep 25, 2024 by carlosgmartin
7
stop_gradient cannot safeguard io_callback against jax.grad transformation enhancement New feature or request
#23911 opened Sep 25, 2024 by ysngshn
Add override_type_lowerings to LoweringParameters enhancement New feature or request
#23904 opened Sep 25, 2024 by erick-xanadu
Implementation of PowerLaw distribution enhancement New feature or request
#23826 opened Sep 22, 2024 by Qazalbash
@jakevdp
[DOC] We should document all jax config from the file config.py documentation enhancement New feature or request
#23809 opened Sep 20, 2024 by nouiz
2 tasks done
Numerical implementation of logarithmic integral function 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
#23732 opened Sep 18, 2024 by Qazalbash
3
Typing improvement: Preserve the wrapped callable's signature in jax.jit enhancement New feature or request
#23719 opened Sep 18, 2024 by lebrice
2 tasks done
1
Document if differentiability of implementation is dependent on hardware enhancement New feature or request
#23650 opened Sep 16, 2024 by unalmis
5
Make jax.distributed timeouts configurable via jax.config enhancement New feature or request
#23599 opened Sep 12, 2024 by skye
1
2
Allow tuple inputs to scatter_dimension in jax.lax.psum_scatter enhancement New feature or request
#23594 opened Sep 12, 2024 by vvvm23
Sparse matrix multiplication and output nse values enhancement New feature or request
#23576 opened Sep 11, 2024 by joaospinto
7
fp32 or even fp64 support for cudnn flashattention enhancement New feature or request
#23560 opened Sep 11, 2024 by wangleiphy
1
Project Plan Github Like Pytorch for features and next releases? enhancement New feature or request
#23537 opened Sep 10, 2024 by johnnynunez
@mattjj
1
add support for QR decomposition gradient calculation for wide matrices (rows < columns) enhancement New feature or request
#23533 opened Sep 10, 2024 by rlucas7
2 tasks done
ProTip! no:milestone will show everything without a milestone.