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
69 Open 378 Closed
69 Open 378 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

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
[WIP] Amend documentation of JAX config options, include context managers, mechanism to hide flags documentation
#28006 opened Apr 14, 2025 by ZacCranko • Draft
@ZacCranko
5
Clarify whether lax.top_k returns results in sorted order documentation
#27594 opened Mar 30, 2025 by carlosgmartin
@jakevdp
4
jax.debug.print and jax.debug.breakpoint docs should mention perturbing the computation documentation
#26370 opened Feb 6, 2025 by mattjj
1
@alhridoy
1
Document jax.lax APIs documentation
#26219 opened Jan 30, 2025 by jakevdp
@jakevdp
Rework comparisons in the docs of JAX vs. NumPy PRNG documentation
#24927 opened Nov 16, 2024 by emilyfertig
1
Reorganize the tutorials documentation type:performance
#24632 opened Oct 30, 2024 by emilyfertig
5 of 16 tasks
@emilyfertig
2
[docs][FR] document how to build asan jaxlib documentation enhancement New feature or request
#24133 opened Oct 4, 2024 by skye
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
[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
Document Named Collectives documentation
#23024 opened Aug 12, 2024 by lockwo
@jakevdp
2
Figure out + document how to use manual capture profiling with Kubernetes documentation enhancement New feature or request
#22096 opened Jun 25, 2024 by skye
1
Pallas/Mosiac documentation and examples documentation enhancement New feature or request pallas Issues pertaining to Pallas (GPU or TPU)
#21337 opened May 21, 2024 by PrajP
@sharadmv @apaszke
2
improve error message and docs for custom_jvp / custom_vjp nondiff_argnums escaped tracer error better_errors Improve the error reporting documentation
#20889 opened Apr 23, 2024 by mattjj
@mattjj
[README] nn libraries fix documentation
#20745 opened Apr 14, 2024 by keren9898
30
in shard_map docs, bidirectional collective matmul, use pad-and-add pattern before dot documentation
#20243 opened Mar 14, 2024 by mattjj
@mattjj
Better doc for jax.clear_caches() documentation question Questions for the JAX team
#19590 opened Jan 30, 2024 by salcianu
@mattjj
1
Improve jax.Array documentation documentation enhancement New feature or request
#19342 opened Jan 12, 2024 by hawkinsp
@yashk2810
2
Tracking issue: Documentation update documentation
#18585 opened Nov 17, 2023 by jakevdp
18 of 26 tasks
2
@jakevdp
2
Add example of jaxpr pretty_print(source_info=True) to the make_jaxpr documentation documentation enhancement New feature or request
#18240 opened Oct 23, 2023 by fcole
@froystig
add jax_debug_infs to docs documentation
#17722 opened Sep 21, 2023 by mattjj
2
Add docstring to PositionalSharding documentation enhancement New feature or request
#16363 opened Jun 12, 2023 by yashk2810
2
@mattjj
1
Document JAX_LOG_COMPILES documentation enhancement New feature or request
#15587 opened Apr 13, 2023 by hawkinsp
@mattjj
1
jax.tree_utils.tree_map raises errors for dictionary with mixed types of keys better_errors Improve the error reporting documentation
#15358 opened Apr 1, 2023 by zi-w
3
Document behaviour of mixed derivatives documentation
#15082 opened Mar 19, 2023 by Jesperoka
@jakevdp
2
ProTip! Mix and match filters to narrow down what you’re looking for.