mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 02:46:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs
History
Hyeontaek Lim 4f8881539d Make the transfer guard documentation easier to find 
Move the main documentation of the transfer guard from "Notes" to "Reference documentation" section for better visibility.

Add a link to the main documentation to the docstring of jax.transfer_guard(), which currently shows up as the top result when searching for "jax transfer_guard".
2022-06-09 18:13:03 +00:00
..
_static
Add an option to create a perfetto link in the JAX profiler
2022-06-01 15:48:29 -07:00
_templates
Add the JAX logo to sphinx docs
2019-08-05 12:34:29 -07:00
design_notes
correct spelling on comment
2022-06-06 15:24:41 +09:00
jax-101
correct spelling on comment
2022-06-06 15:24:41 +09:00
notebooks
Rename mesh into Mesh in xmap tutorial and doc
2022-06-03 17:11:09 +07:00
sphinxext
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
api_compatibility.md
Add an API compatibility policy.
2022-03-01 14:28:22 -05:00
async_dispatch.rst
Add the changes suggested
2021-08-01 20:13:02 -07:00
autodidax.ipynb
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
autodidax.md
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
autodidax.py
feat: refactor code using pyupgrade
2022-05-17 22:14:05 +01:00
changelog.md
move changelog to top level
2021-03-08 10:44:52 -08:00
concurrency.rst
Document that JAX follows the NEP-29 deprecation policy.
2021-04-21 11:12:41 -04:00
conf.py
DOC: fix docs generator to group notebook code output
2022-05-21 09:36:38 +02:00
contributing.md
Remove .[minimum-jaxlib] from test-requirements.txt
2021-09-23 12:24:24 -07:00
custom_vjp_update.md
DOC: many small fixes
2021-08-04 16:55:13 -07:00
deprecation.md
Document that JAX follows the NEP-29 deprecation policy.
2021-04-21 11:12:41 -04:00
developer.md
DOC: update myst-nb to v0.15.0
2022-05-09 14:32:48 -07:00
device_memory_profiling.md
DOC: many small fixes
2021-08-04 16:55:13 -07:00
errors.rst
Expose UnexpectedTracerError and add docs.
2021-07-27 23:23:28 +01:00
faq.rst
Merge pull request #9118 from skye:device_context_manager
2022-06-02 10:33:53 -07:00
glossary.rst
Fix DeviceArray class reference
2022-01-10 17:34:09 +03:00
gpu_memory_allocation.rst
Update gpu_memory_allocation.rst
2021-02-13 15:47:42 +02:00
index.rst
Make the transfer guard documentation easier to find
2022-06-09 18:13:03 +00:00
installation.rst
DOC: link to install instructions in HTML docs
2022-04-22 12:35:31 -07:00
jax_internal_api.rst
Add jax.errors submodule & error troubleshooting docs
2021-03-03 12:39:12 -08:00
jax.config.rst
Add jax.default_device context manager
2022-05-07 00:31:00 +00:00
jax.distributed.rst
Add jax.distributed and jax.dlpack to the docs.
2022-02-17 16:10:07 -05:00
jax.dlpack.rst
Add jax.distributed and jax.dlpack to the docs.
2022-02-17 16:10:07 -05:00
jax.example_libraries.optimizers.rst
[JAX] move example libraries from jax.experimental into jax.example_libraries
2021-10-19 17:30:45 -07:00
jax.example_libraries.rst
Documentation improvements.
2022-03-08 09:37:33 -05:00
jax.example_libraries.stax.rst
[JAX] move example libraries from jax.experimental into jax.example_libraries
2021-10-19 17:30:45 -07:00
jax.experimental.global_device_array.rst
Fix the display of GDA docs
2022-03-03 12:19:04 -08:00
jax.experimental.host_callback.rst
DOC: move experimental APIs to their own pages
2022-02-04 14:40:34 -08:00
jax.experimental.jet.rst
Wrote docstring for jet
2022-02-12 08:14:57 +01:00
jax.experimental.loops.rst
An implementation of an experimental syntactic sugar for 'for' loops.
2019-11-16 17:23:40 +01:00
jax.experimental.maps.rst
Add a docstring for maps.Mesh
2022-04-15 13:46:59 -07:00
jax.experimental.pjit.rst
Add a docstring for pjit
2021-05-06 12:00:49 -07:00
jax.experimental.rst
Merge pull request #9546 from pnkraemer:jet-docs
2022-03-02 18:04:40 -08:00
jax.experimental.sparse.rst
[sparse] incremental improvement to docs
2022-05-04 12:25:16 -07:00
jax.flatten_util.rst
add ravel_pytree to generated docs
2021-08-02 17:55:41 -07:00
jax.image.rst
Documentation improvements.
2022-03-08 09:37:33 -05:00
jax.lax.rst
DOC: add missing linalg functionality to docs
2022-03-15 09:55:59 -07:00
jax.lib.rst
Remove CpuDevice from documentation index.
2022-04-16 10:03:47 -04:00
jax.nn.initializers.rst
Document jax.nn.initializers.
2022-03-07 17:26:04 -05:00
jax.nn.rst
Alphabetize docs.
2020-10-15 11:10:56 -04:00
jax.numpy.rst
DeviceArray: Improve support for copy, deepcopy, and pickle
2022-05-19 12:00:58 -07:00
jax.ops.rst
Remove jax.ops.index... functions.
2022-02-24 09:36:28 -05:00
jax.profiler.rst
docs: tweaks to make docs build on Python 3.8
2021-04-01 15:50:15 -07:00
jax.random.rst
Added random.generalized_normal and random.ball.
2022-06-03 15:11:29 -04:00
jax.rst
Add jax.distributed and jax.dlpack to the docs.
2022-02-17 16:10:07 -05:00
jax.scipy.rst
Add jax.scipy.linalg.funm
2022-05-02 21:46:41 +01:00
jax.tree_util.rst
unify tree_map and tree_multimap
2021-04-28 19:59:31 -07:00
jaxpr.rst
DOC: add references for haskell-style signatures
2022-05-13 12:35:28 -07:00
multi_process.md
DOC: many small fixes
2021-08-04 16:55:13 -07:00
profiling.md
correct spelling on comment
2022-06-06 15:24:41 +09:00
pytrees.md
DOC: add section on input validation in custom pytrees
2022-04-13 13:50:05 -07:00
rank_promotion_warning.rst
DOC: many small fixes
2021-08-04 16:55:13 -07:00
README.md
Fixes to the documentation
2019-10-09 17:45:09 +02:00
requirements.txt
CI: unpin traitlets
2022-06-08 11:57:10 -07:00
transfer_guard.rst
Add a documentation for the JAX transfer guard
2022-03-25 19:50:12 +00:00
transformations.md
CI: update jupytext to 0.13.8
2022-05-02 11:22:31 -07:00
type_promotion.rst
Fix nomenclature in type promotion doc
2022-02-15 13:26:06 -08:00

README.md

To rebuild the documentation, see Update Documentation.