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
86 Open 6,104 Closed
86 Open 6,104 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

Clean up conditional lowering of tan after JAX v0.4.34 release. pull ready Ready for copybara import and testing
#24405 opened Oct 19, 2024 by dfm Loading…
@dfm
Better documentation for jnp.load documentation pull ready Ready for copybara import and testing
#24403 opened Oct 19, 2024 by jakevdp Loading…
@jakevdp
4
[export] Add support for serialization for some custom PyTree nodes pull ready Ready for copybara import and testing
#24342 opened Oct 16, 2024 by gnecula Loading…
@gnecula
17
fix liveness issues within jax itself pull ready Ready for copybara import and testing
#24313 opened Oct 15, 2024 by mattjj Loading…
@mattjj
Re-enable cudnn_fusion_test on A100. pull ready Ready for copybara import and testing
#24069 opened Oct 2, 2024 by sergachev Loading…
@dfm
4
improve readthedocs behavior for jax.remat / jax.checkpoint documentation pull ready Ready for copybara import and testing
#24064 opened Oct 2, 2024 by mattjj Loading…
@mattjj
2
[Mosaic GPU] Query SM and PTX ISA dynamically using driver and LLVM pull ready Ready for copybara import and testing
#24021 opened Sep 30, 2024 by andportnoy Loading…
3
Improve typing of jax.jit pull ready Ready for copybara import and testing
#23720 opened Sep 18, 2024 by lebrice Loading…
@jakevdp
39
Add float8_e4m3 and float8_e3m4 types support pull ready Ready for copybara import and testing
#23585 opened Sep 12, 2024 by apivovarov Loading…
@jakevdp
54
Determine symmetric linear op fro CG from abstract output dtype pull ready Ready for copybara import and testing
#23486 opened Sep 6, 2024 by Joshuaalbert Loading…
@jakevdp
21
Adding JAX_LOGGING_LEVEL configuration option pull ready Ready for copybara import and testing
#23468 opened Sep 6, 2024 by rdyro Loading…
@skye
47
Stackless [WIP] pull ready Ready for copybara import and testing
#23299 opened Aug 28, 2024 by dougalm • Draft
When caching is enabled, also enable XLA caching features as well pull ready Ready for copybara import and testing
#22899 opened Aug 6, 2024 by trevor-m Loading…
@dfm
17
Make scatter mode default to PROMISE_IN_BOUNDS pull ready Ready for copybara import and testing
#22801 opened Jul 31, 2024 by jakevdp • Draft
@jakevdp
3
[shard-map] relax vmap-of-shmap error with spmd_axis_name pull ready Ready for copybara import and testing
#22676 opened Jul 25, 2024 by mattjj • Draft
improve stop_gradient(mutable_array) error pull ready Ready for copybara import and testing
#22521 opened Jul 18, 2024 by mattjj Loading…
3
Add #22244 (remat numerics fix/possibly breaking change) to CHANGELOG pull ready Ready for copybara import and testing
#22424 opened Jul 12, 2024 by skye Loading…
Delete jax/tools/colab_tpu.py pull ready Ready for copybara import and testing
#22401 opened Jul 11, 2024 by sagelywizard Loading…
@jakevdp
4
[pallas] Add tests for shape polymorphism for TPU pull ready Ready for copybara import and testing
#22060 opened Jun 24, 2024 by gnecula Loading…
@gnecula
Pallas bitwise_left_shift unit test fix pull ready Ready for copybara import and testing
#21780 opened Jun 10, 2024 by rahulbatra85 Loading…
4
relax custom_vjp dtype checks, tweak float0 behavior pull ready Ready for copybara import and testing
#21537 opened May 31, 2024 by mattjj • Draft
print literal types in jaxprs pull ready Ready for copybara import and testing
#21415 opened May 24, 2024 by mattjj Loading…
Add boundary='wrap' support for jax.scipy.signal.convolve2d and jax.scipy.signal.correlate2d pull ready Ready for copybara import and testing
#21241 opened May 15, 2024 by rajasekharporeddy Loading…
1
@jakevdp
32
Add new flash attn features to cuDNN SDPA API and remove fused attn pull ready Ready for copybara import and testing
#21228 opened May 14, 2024 by Cjkkkk Loading…
@superbobry
19
lobpcg: avoid changing X64 setting pull ready Ready for copybara import and testing
#21165 opened May 10, 2024 by jakevdp Loading…
@jakevdp
ProTip! Adding no:label will show everything without a label.