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
Peter Hawkins e42e52d4aa Rename test flag --num_generated_cases to --jax_num_generated_cases. 
parse_flags_with_absl() only parses flags that start with --jax_. Other flags are only parsed when absl.app's main function runs. But that's too late for test cases: test cases need to have the number of generated cases chosen at module initialization time. Hence the --num_generated_cases flag wasn't doing anything. Oops. By renaming it it works once again.

It might make sense to stop using flags for the number of generated cases and only use environment variables. We defer that to a future change.

Fix many test cases that were shown to be broken with a larger number of test cases enabled.

PiperOrigin-RevId: 487406670
2022-11-09 18:58:05 -08:00
..
_static
JEP 12049: Type Annotation Roadmap
2022-09-13 09:14:48 -07:00
_templates
Add the JAX logo to sphinx docs
2019-08-05 12:34:29 -07:00
debugging
Fix typo in jax.debug.print() documentation.
2022-08-30 14:51:12 -07:00
jax-101
Clarify in JAX Basics that JAX array creation is also an operation that requires accelerator dispatch and converting to a regular Python type is a blocking operation.
2022-10-26 13:38:17 -04:00
jep
Fix typo in 9419-jax-versioning.md
2022-10-25 03:26:48 +09:00
notebooks
Remove a TODO from the xmap tutorial
2022-10-26 15:44:06 +00:00
sphinxext
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
aot.md
write in-process AOT walkthrough doc
2022-09-02 13:02:25 -07:00
api_compatibility.md
Deprecate top-level aliases of jax.tree_util functions
2022-07-07 11:41:46 -07:00
async_dispatch.rst
Add the changes suggested
2021-08-01 20:13:02 -07:00
autodidax.ipynb
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
autodidax.md
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
autodidax.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -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
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
contributing.md
Remove .[minimum-jaxlib] from test-requirements.txt
2021-09-23 12:24:24 -07:00
deprecation.md
Document that JAX follows the NEP-29 deprecation policy.
2021-04-21 11:12:41 -04:00
developer.md
Rename test flag --num_generated_cases to --jax_num_generated_cases.
2022-11-09 18:58:05 -08:00
device_memory_profiling.md
DOC: update pprof install instructions
2022-06-21 11:15:14 +02:00
errors.rst
Expose UnexpectedTracerError and add docs.
2021-07-27 23:23:28 +01:00
faq.rst
FAQ: fix JIT numerics discussion
2022-10-26 11:30:17 -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
Increase visibility in index Multi-Host Multi-Process Envs Guide
2022-11-04 19:22:39 +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.debug.rst
Update debugging docs to have sharding visualization
2022-09-22 19:42:36 -07:00
jax.distributed.rst
Document multiprocess GPU support.
2022-08-09 11:31:05 -04: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.checkify.rst
Add debugging tools under jax.debug and documentation
2022-07-28 20:07:26 -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.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
Add debugging tools under jax.debug and documentation
2022-07-28 20:07:26 -07: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
Add support for Hessenberg and tridiagonal matrix reductions on CPU.
2022-11-09 06:23:55 -08:00
jax.lib.rst
Bump JAX/Jaxlib versions
2022-06-28 14:36:47 -07:00
jax.nn.initializers.rst
Document jax.nn.initializers.
2022-03-07 17:26:04 -05:00
jax.nn.rst
update docs to point to jax.nn.standardize
2022-10-20 21:25:37 -07:00
jax.numpy.rst
Point to ndarray.at from docstring of unimplemented jnp.put & jnp.place
2022-10-28 14:13:36 -07:00
jax.ops.rst
Remove jax.ops.index... functions.
2022-02-24 09:36:28 -05:00
jax.profiler.rst
jax.profiler: remove deprecated functions
2022-07-19 08:13:44 -07:00
jax.random.rst
Added random.generalized_normal and random.ball.
2022-06-03 15:11:29 -04:00
jax.rst
Add jax.print_environment_info()
2022-09-12 15:39:33 -07:00
jax.scipy.rst
Add support for Hessenberg and tridiagonal matrix reductions on CPU.
2022-11-09 06:23:55 -08:00
jax.stages.rst
write and generate package API documentation for jax.stages
2022-09-01 19:26:53 -07: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
Add generic interface for auto initialization of distributed JAX service
2022-09-22 14:15:38 -05: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: add absl-py to docs/requirements.txt
2022-11-04 11:28:15 -07:00
transfer_guard.rst
Add a documentation for the JAX transfer guard
2022-03-25 19:50:12 +00:00
transformations.md
pre-commit hook: update flake8, mypy, & jupytext
2022-08-15 15:32:45 -07:00
type_promotion.rst
Create JAX Enhancement Proposals (JEPs).
2022-08-08 16:13:58 -04:00

README.md

To rebuild the documentation, see Update Documentation.