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
73 Open 309 Closed
73 Open 309 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

Better docs for jnp.fromfunction documentation
#24412 opened Oct 20, 2024 by jakevdp Loading…
@jakevdp
Better documentation for jnp.load documentation pull ready Ready for copybara import and testing
#24403 opened Oct 19, 2024 by jakevdp Loading…
@jakevdp
4
Make jnp.subtract a ufunc documentation
#24400 opened Oct 19, 2024 by jakevdp Loading…
@jakevdp
[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
Update installation instructions documentation
#22410 opened Jul 12, 2024 by garymm Loading…
1
@hawkinsp
6
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
Tracking issue: inline docstrings documentation
#21461 opened May 28, 2024 by jakevdp
@jakevdp
2
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
Documentation example for enabling x64 mode shows x32 output bug Something isn't working documentation
#17273 opened Aug 24, 2023 by maximilianruesch
4
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
2
Document behaviour of mixed derivatives documentation
#15082 opened Mar 19, 2023 by Jesperoka
@jakevdp
2
NCCL and compatibility with AWS EFA documentation enhancement New feature or request NVIDIA GPU Issues specific to NVIDIA GPUs
#14534 opened Feb 16, 2023 by mathemakitten
2
ProTip! Updated in the last three days: updated:>2024-10-17.