mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs
History
Peter Hawkins e558feaa5e Deprecate support for the mhlo dialect. 
JAX has not used mhlo for some time, in favor of stablehlo. Deprecate support for this dialect in JAX's API and remove testing.

PiperOrigin-RevId: 598550225
2024-01-15 02:13:40 -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
CI: update jupytext version
2024-01-09 14:34:21 -08:00
jep
Deprecate support for the mhlo dialect.
2024-01-15 02:13:40 -08:00
notebooks
CI: update jupytext version
2024-01-09 14:34:21 -08:00
pallas
Deprecate support for the mhlo dialect.
2024-01-15 02:13:40 -08:00
sphinxext
DOC: specify jax sphinx extension as safe for parallel read/write
2022-12-19 15:06:28 -08:00
tutorials
DOC: add a note about installing older GPU wheels
2024-01-12 11:12:39 -08: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
Fix duplicate word occurrences
2023-12-19 06:15:30 +01:00
autodidax.md
CI: update jupytext version
2024-01-09 14:34:21 -08:00
autodidax.py
CI: update jupytext version
2024-01-09 14:34:21 -08: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
CI: update jupytext version
2024-01-09 14:34:21 -08: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
Fix duplicate word occurrences
2023-12-19 06:15:30 +01:00
index.rst
Improve installation instructions.
2023-09-22 17:13:47 -04:00
installation.md
DOC: add a note about installing older GPU wheels
2024-01-12 11:12:39 -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
Cleanup deprecated compilation cache APIs.
2024-01-12 22:44:48 -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 jax.numpy.concat
2024-01-12 13:12:09 -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
Remove a number of deprecated APIs
2023-12-18 10:08:47 -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
Merge pull request #19092 from 8bitmp3:jax-docs-gradient-checkpointing
2024-01-09 13:50:13 -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.