mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 20:36:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs
History
Peter Hawkins 3e5ecfe363 Add jax.distributed and jax.dlpack to the docs. 
Reorder the doc modules into something closer to alphabetical order.

Add missing functions from jax.scipy.linalg and jax.scipy.signal to the docs.
2022-02-17 16:10:07 -05:00
..
_static
Fix nomenclature in type promotion doc
2022-02-15 13:26:06 -08:00
_templates
Add the JAX logo to sphinx docs
2019-08-05 12:34:29 -07:00
design_notes
type promotion design doc: minor typos
2022-02-15 12:17:18 -08:00
jax-101
maint: update pre-commit package versions & fix new mypy errors
2022-01-31 13:39:11 -08:00
notebooks
maint: update pre-commit package versions & fix new mypy errors
2022-01-31 13:39:11 -08:00
sphinxext
[JAX] Fix pylint errors.
2021-10-04 17:54:46 -07:00
async_dispatch.rst
Add the changes suggested
2021-08-01 20:13:02 -07:00
autodidax.ipynb
Consolidate more XLA-lowering logic between jit, pmap, and xmap.
2021-11-30 14:24:33 -08:00
autodidax.md
maint: update pre-commit package versions & fix new mypy errors
2022-01-31 13:39:11 -08:00
autodidax.py
maint: update pre-commit package versions & fix new mypy errors
2022-01-31 13:39:11 -08: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
Add type promotion design doc
2022-02-14 14:16:42 -08: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
Add -j auto to the suggested doc build instructions.
2022-02-04 08:57:14 -05: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
Add more documentation for buffer donation
2022-01-24 09:33:08 +01: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
Move design notes into docs/, and render them as part of the documentation.
2022-02-02 14:29:03 -05:00
jax_internal_api.rst
Add jax.errors submodule & error troubleshooting docs
2021-03-03 12:39:12 -08:00
jax.config.rst
DOC: add documentation for configuration functionality
2021-11-29 10:44:54 -08: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
[JAX] move example libraries from jax.experimental into jax.example_libraries
2021-10-19 17:30:45 -07: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.ann.rst
DOC: move experimental APIs to their own pages
2022-02-04 14:40:34 -08:00
jax.experimental.global_device_array.rst
DOC: move experimental APIs to their own pages
2022-02-04 14:40:34 -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.loops.rst
An implementation of an experimental syntactic sugar for 'for' loops.
2019-11-16 17:23:40 +01:00
jax.experimental.maps.rst
DOC: move experimental APIs to their own pages
2022-02-04 14:40:34 -08:00
jax.experimental.pjit.rst
Add a docstring for pjit
2021-05-06 12:00:49 -07:00
jax.experimental.rst
Checkify: rename some symbols and add some docstrings.
2022-02-08 17:40:04 +00:00
jax.experimental.sparse.rst
DOC: move experimental APIs to their own pages
2022-02-04 14:40:34 -08:00
jax.flatten_util.rst
add ravel_pytree to generated docs
2021-08-02 17:55:41 -07:00
jax.image.rst
Expose scale_and_translate as a public function, fix a bug in implementation when translation is not 0.
2020-09-18 13:30:58 -07:00
jax.lax.rst
Merge branch 'main' into conv_local
2022-01-07 09:51:46 -08:00
jax.lib.rst
Remove xla_bridge.{constant, register_constant_handler, _python_scalar_constant} from API.
2021-10-18 13:56:58 -07:00
jax.nn.initializers.rst
add docs for jax.nn
2019-08-29 18:15:36 -07:00
jax.nn.rst
Alphabetize docs.
2020-10-15 11:10:56 -04:00
jax.numpy.rst
[JAX] Update users of jax.ops.index... functions, which are deprecated.
2021-10-28 09:54:26 -07:00
jax.ops.rst
DOC: add info about repeated indices to jax.ops docs
2022-01-27 10:22:25 -08: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
DOC: fix jax.random documentation
2020-11-20 16:41:32 -08: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.distributed and jax.dlpack to the docs.
2022-02-17 16:10:07 -05:00
jax.tree_util.rst
unify tree_map and tree_multimap
2021-04-28 19:59:31 -07:00
jaxpr.rst
custom pp_eqn rules, simpler xla_call print
2021-11-23 15:52:52 -08:00
multi_process.md
DOC: many small fixes
2021-08-04 16:55:13 -07:00
profiling.md
Remove documentation for --xla_hlo_profile.
2021-10-06 09:43:32 -04:00
pytrees.md
DOC: many small fixes
2021-08-04 16:55:13 -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
Release jax and jaxlib 0.3.0 as per the new release process.
2022-02-10 11:59:13 -08:00
transformations.md
maint: update pre-commit package versions & fix new mypy errors
2022-01-31 13:39:11 -08: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.