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
106 Open 7,012 Closed
106 Open 7,012 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
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

[typing] adjust axis annotation for ufunc.reduce pull ready Ready for copybara import and testing
#29120 opened May 30, 2025 by jakevdp Loading…
1
@jakevdp
[scan] don't hoist loop-invariant computations in scan, just forward kokoro:force-run pull ready Ready for copybara import and testing
#28993 opened May 23, 2025 by mattjj Loading…
@mattjj
4
[JAX][DOC] Add optimizer state offloading doc kokoro:force-run pull ready Ready for copybara import and testing
#28988 opened May 23, 2025 by zhenying-liu Loading…
10
Use the builtin isinstance with MLIR types pull ready Ready for copybara import and testing
#28973 opened May 23, 2025 by superbobry Loading…
[jax2tf] Refine the disabling of jax2tf_test, for versions <= 2.19.1 kokoro:force-run pull ready Ready for copybara import and testing
#28916 opened May 21, 2025 by gnecula Loading…
@gnecula
Experiment: mutable RNG object pull ready Ready for copybara import and testing
#28845 opened May 20, 2025 by jakevdp • Draft
@jakevdp
added solve_sylvester and accompanying tests kokoro:force-run pull ready Ready for copybara import and testing
#28810 opened May 17, 2025 by Dekermanjian Loading…
@jakevdp
54
Deprecate arr.T property for arrays with ndim != 2 kokoro:force-run pull ready Ready for copybara import and testing
#28798 opened May 16, 2025 by jakevdp • Draft
@jakevdp
1
fix deprecation warning for jax.lax infeed/outfeed kokoro:force-run pull ready Ready for copybara import and testing
#28767 opened May 15, 2025 by jakevdp Loading…
@jakevdp
2
Ensure __jax_array__ support in binary ops kokoro:force-run pull ready Ready for copybara import and testing
#28630 opened May 8, 2025 by jakevdp Loading…
@jakevdp
fix box bug in scan transpose kokoro:force-run pull ready Ready for copybara import and testing
#28561 opened May 6, 2025 by mattjj Loading…
@mattjj
Remove broken jax/jaxlib version test. pull ready Ready for copybara import and testing
#28515 opened May 5, 2025 by hawkinsp Loading…
Fix overloaded type signature for jax.numpy.where. pull ready Ready for copybara import and testing
#28314 opened Apr 27, 2025 by carlosgmartin Loading…
@superbobry
10
Move DropVar annotation into Jaxpr pretty printer pull ready Ready for copybara import and testing
#28145 opened Apr 21, 2025 by dfm Loading…
@dfm
[Pallas] [Mosaic GPU] Add GPU pipelining docs kokoro:force-run pull ready Ready for copybara import and testing
#28135 opened Apr 19, 2025 by justinjfu Loading…
@apaszke
83
Add jax_logging_level to common config options in docs documentation kokoro:force-run pull ready Ready for copybara import and testing
#28034 opened Apr 15, 2025 by skye Loading…
5
jax.jit: support decorator factory pattern kokoro:force-run pull ready Ready for copybara import and testing
#28011 opened Apr 14, 2025 by jakevdp Loading…
1
@jakevdp
12
Fix jax.random.randint for 8-bit and 16-bit dtypes kokoro:force-run pull ready Ready for copybara import and testing
#27742 opened Apr 4, 2025 by jakevdp Loading…
@jakevdp
4
Fix shard_map for primitives with multiple outputs and None specs kokoro:force-run pull ready Ready for copybara import and testing
#27720 opened Apr 3, 2025 by justinjfu Loading…
13
Remove some unnecessary dtype canonicalizations. pull ready Ready for copybara import and testing
#27334 opened Mar 21, 2025 by dfm Loading…
@dfm
Fix wrong unpacking in the decode attention pallas kernel kokoro:force-run pull ready Ready for copybara import and testing
#27292 opened Mar 20, 2025 by uuirs Loading…
2
jax.nn.initializers: use Sequence[int] directly for shape annotations. kokoro:force-run pull ready Ready for copybara import and testing
#27202 opened Mar 17, 2025 by jakevdp Loading…
1
@jakevdp
1
Add a custom_partitioning helper to support batch partitioning pull ready Ready for copybara import and testing
#27067 opened Mar 11, 2025 by dfm Loading…
@dfm
[pallas:triton] Fix atomic min/max lowering for uint and float types pull ready Ready for copybara import and testing
#26604 opened Feb 19, 2025 by draganmladjenovic Loading…
5
[better_errors] Simplify api_util.debug_info. pull ready Ready for copybara import and testing
#26603 opened Feb 19, 2025 by gnecula Loading…
@gnecula
6
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.