mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 12:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs
History
Skye Wanderman-Milne 5d26c307ce Revert "Recommend the plugin in the CUDA installation instructions." 
This reverts commit 78bc4f5ced41f7f96a70da769e7c3170dc2b3161.

GPU dlpack is broken with the new plugin. Recommend working GPU jaxlib until we can fix.
2023-12-14 17:23:18 -08:00
..
_static
JEP 9263: Typed keys & pluggable RNGs
2023-09-20 11:26:13 -07:00
_templates
Add the JAX logo to sphinx docs
2019-08-05 12:34:29 -07:00
debugging
DOC: mention static values in debug.print docs
2023-08-28 10:59:40 -07:00
jax-101
Correcting typo in docs, 06-parallelism.
2023-11-28 01:34:49 -08:00
jep
JAX tutorials: pseudorandom numbers
2023-11-21 14:26:06 -08:00
notebooks
Fix wording in TFDS example
2023-11-03 11:31:48 -07:00
pallas
Ah, jupytext converts from markdown to a notebook, not vice versa.
2023-11-09 11:11:21 -05:00
sphinxext
DOC: specify jax sphinx extension as safe for parallel read/write
2022-12-19 15:06:28 -08:00
tutorials
Update thinking-in-jax working-with-pytrees
2023-12-13 22:57:46 +00:00
advanced_guide.rst
fixes from reviewers
2023-02-23 15:06:55 -08:00
aot.md
AOT: better error messages on call signature mismatch
2023-07-10 22:10:50 -07:00
api_compatibility.md
add jax.prng to uncovered modules list in API policy
2023-07-18 14:13:25 -07:00
async_dispatch.rst
Asynchronous dispatch doc update regarding jax.Array migration
2023-10-05 13:45:51 -07:00
autodidax.ipynb
Replace references to DeviceArray with Array.
2023-08-18 17:46:00 -04:00
autodidax.md
CI: update pre-commit hooks to latest version
2023-10-30 09:12:24 -07:00
autodidax.py
CI: update pre-commit hooks to latest version
2023-10-30 09:12:24 -07:00
beginner_guide.rst
DOC: more guidance within getting started guide
2023-01-25 12:00:21 -08:00
building_on_jax.md
Add building on JAX
2023-05-22 10:05:39 -07: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
CI: fix doc build
2023-11-16 07:59:07 -08:00
contributing.md
Switch from flake8 to Ruff
2023-11-15 22:35:52 -05:00
contributor_guide.rst
Add doc about regression investigation.
2023-09-29 09:51:49 -04:00
Custom_Operation_for_GPUs.md
Fix the tutorial
2023-11-05 08:17:23 -08:00
deprecation.md
Change deprecation policy from NEP 29 to SPEC 0
2023-11-06 11:40:15 -08:00
developer.md
Squashed commit - documentation matplotlib
2023-11-28 11:36:50 +01:00
device_memory_profiling.md
Update profiling docs.
2023-05-30 14:27:11 -07:00
errors.rst
errors: create TracerBoolConversionError for more targeted debugging tips
2023-06-21 01:41:45 -07:00
faq.rst
Deprecate the device() method of JAX arrays
2023-11-30 11:43:02 -08:00
glossary.rst
Fix DeviceArray class reference
2022-01-10 17:34:09 +03:00
gpu_memory_allocation.rst
Change default memory allocation to 75% instead of 90%
2023-07-13 14:44:38 -07:00
gpu_performance_tips.md
draft docs on gpu performance tuning
2023-10-30 20:33:56 -07:00
index.rst
Improve installation instructions.
2023-09-22 17:13:47 -04:00
installation.md
Revert "Recommend the plugin in the CUDA installation instructions."
2023-12-14 17:23:18 -08:00
investigating_a_regression.md
Add a comment
2023-11-02 11:37:17 -07:00
jax_array_migration.md
Fixes broken examples, and (invalid) comment for PartitionSpec
2023-03-17 16:09:45 -07:00
jax_internal_api.rst
Add jax.errors submodule & error troubleshooting docs
2021-03-03 12:39:12 -08:00
jax.debug.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.distributed.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.dlpack.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.dtypes.rst
custom prng: introduce mechanism to identify key arrays by dtype
2023-07-21 12:27:32 -07:00
jax.example_libraries.optimizers.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.example_libraries.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.example_libraries.stax.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.experimental.array_api.rst
Add jax.experimental.array_api interface
2023-11-16 14:21:04 -08:00
jax.experimental.checkify.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.experimental.compilation_cache.rst
Document the compilation cache
2023-11-20 07:03:30 -08:00
jax.experimental.custom_partitioning.rst
Adding @custom_partitioning to jax.experimental API doc
2023-01-20 11:34:29 -08:00
jax.experimental.host_callback.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.experimental.jet.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.experimental.key_reuse.rst
Add experimental static key reuse checking
2023-12-11 12:03:48 -08:00
jax.experimental.maps.rst
Merge pull request #14248 from jakevdp:dead-code
2023-02-01 21:25:46 +00:00
jax.experimental.multihost_utils.rst
Fix a sphinx error.
2023-05-05 11:08:18 -07:00
jax.experimental.pjit.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.experimental.rst
Add experimental static key reuse checking
2023-12-11 12:03:48 -08:00
jax.experimental.sparse.rst
DOC: document jax.experimental.sparse.linalg
2023-04-28 14:18:50 -07:00
jax.extend.linear_util.rst
Add jax.extend.mlir.
2023-12-06 09:16:43 -08:00
jax.extend.mlir.rst
Add jax.extend.mlir.
2023-12-06 09:16:43 -08:00
jax.extend.random.rst
Add jax.extend.mlir.
2023-12-06 09:16:43 -08:00
jax.extend.rst
Add jax.extend.mlir.
2023-12-06 09:16:43 -08:00
jax.flatten_util.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.image.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.lax.rst
Expose PrecisionLike
2023-12-06 14:41:22 -05:00
jax.lib.rst
Export Device as jax.Device.
2023-02-02 12:58:15 -05:00
jax.nn.initializers.rst
Add truncated normal initializer to jax.nn
2023-09-07 16:23:42 -07:00
jax.nn.rst
Add nn.squareplus.
2023-11-14 23:52:41 -05:00
jax.numpy.rst
array api: add jnp.linalg.cross & jnp.linalg.outer
2023-12-12 11:22:31 -08:00
jax.ops.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.profiler.rst
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
jax.random.rst
add random.binomial
2023-11-19 14:51:10 +08:00
jax.rst
add jax.custom_gradient to readthedocs
2023-11-20 13:26:52 -08:00
jax.scipy.rst
Add special.factorial function
2023-12-04 06:13:14 -08:00
jax.sharding.rst
Improve the sharding documentation.
2023-08-03 11:12:02 -04:00
jax.stages.rst
Add a .cost_analysis() on lowered but uncompiled computations.
2023-02-06 12:57:57 -08:00
jax.tree_util.rst
Add jax.tree_util.tree_leaves_with_path(tree).
2023-06-12 04:13:37 -07:00
jax.typing.rst
Define jax.typing.DTypeLike
2023-10-10 08:46:36 -07:00
jaxpr.rst
Print pjit name= before other params
2023-12-07 16:54:07 +00:00
multi_process.md
Add info about iterating over collections with non-deterministic ordering.
2023-04-07 09:12:49 -07:00
notes.rst
DOC: re-organize notes section
2023-01-30 11:57:06 -08:00
profiling.md
Document how to enable concurrent kernel tracing on GPU.
2023-06-16 13:20:27 -04:00
pytrees.md
Copybara import of the project:
2023-03-16 10:00:24 -07:00
rank_promotion_warning.rst
Change to simpler import for jax.config
2023-04-21 11:51:22 -07:00
README.md
Fixes to the documentation
2019-10-09 17:45:09 +02:00
requirements.txt
[export] Add support for serialization and deserialization of Exported
2023-12-11 23:23:02 -08:00
transfer_guard.rst
Add a documentation for the JAX transfer guard
2022-03-25 19:50:12 +00:00
type_promotion.rst
Use jax.Array by default for doctests
2022-11-15 11:52:22 -08:00
user_guides.rst
draft docs on gpu performance tuning
2023-10-30 20:33:56 -07:00

README.md

To rebuild the documentation, see Update Documentation.