mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-24 20:46:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs
History
George Necula a5c3468c93 Added the first draft of the Jaxpr documentation. 
This replaces the previous Google Doc version, and is now
updated with the latest changes in Jaxpr.
2020-02-12 13:01:43 +01:00
..
_static
White background
2019-08-05 13:08:17 -07:00
_templates
Add the JAX logo to sphinx docs
2019-08-05 12:34:29 -07:00
notebooks
Fix Gotchas notebook regarding control flow differentiation. (#2194)
2020-02-10 16:39:27 -08:00
async_dispatch.rst
Fix sphinx references in JAX docs
2019-08-10 16:40:08 -07:00
concurrency.rst
Fix sphinx references in JAX docs
2019-08-10 16:40:08 -07:00
conf.py
Added the first draft of the Jaxpr documentation.
2020-02-12 13:01:43 +01:00
developer.rst
Disabled tests known to fail on Mac, and optionally slow tests.
2020-02-05 18:02:56 +01:00
gpu_memory_allocation.rst
Update GPU memory documentation.
2019-08-21 12:02:42 -07:00
index.rst
Added the first draft of the Jaxpr documentation.
2020-02-12 13:01:43 +01:00
jax_internal_api.rst
Added the first draft of the Jaxpr documentation.
2020-02-12 13:01:43 +01:00
jax.dlpack.rst
Added the first draft of the Jaxpr documentation.
2020-02-12 13:01:43 +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.optimizers.rst
Add segment_sum to the docs and fix its rendering.
2019-05-08 21:18:17 -04:00
jax.experimental.rst
An implementation of an experimental syntactic sugar for 'for' loops.
2019-11-16 17:23:40 +01:00
jax.experimental.stax.rst
Add Sphinx-generated reference documentation for JAX.
2019-01-16 09:13:31 -05:00
jax.experimental.vectorize.rst
Added JAX pytrees notebook
2019-11-24 20:29:07 +01:00
jax.lax.rst
Add gammainc/gammaincc to JAX (#2064)
2020-01-29 11:25:21 -05:00
jax.nn.initializers.rst
add docs for jax.nn
2019-08-29 18:15:36 -07:00
jax.nn.rst
Document functions in jax.nn. (#1795)
2019-12-02 14:21:10 -05:00
jax.numpy.rst
Implement np.block. (#2106)
2020-01-29 11:55:53 -05:00
jax.ops.rst
Add documentation to JAX modules.
2019-07-21 15:55:47 -04:00
jax.random.rst
Added JAX pytrees notebook
2019-11-24 20:29:07 +01:00
jax.rst
Added the first draft of the Jaxpr documentation.
2020-02-12 13:01:43 +01:00
jax.scipy.rst
Add jax.scipy.stats.logistic to documentation. (#2149)
2020-02-03 12:44:57 -05:00
jax.tree_util.rst
Added JAX pytrees notebook
2019-11-24 20:29:07 +01:00
jaxpr.rst
Added the first draft of the Jaxpr documentation.
2020-02-12 13:01:43 +01:00
profiling.rst
Add some brief documentation about how to profile/trace JAX programs.
2019-08-08 21:02:41 -04:00
rank_promotion_warning.rst
tweak rank promotion warning docs
2019-08-25 14:36:23 -07:00
README.md
Fixes to the documentation
2019-10-09 17:45:09 +02:00
requirements.txt
Moved all notebooks to docs/notebooks. (#1493)
2019-10-17 08:58:25 +02:00
type_promotion.rst
Change JAX type promotion to prefer inexact types. (#1815)
2019-12-05 10:57:23 -05:00

README.md

To rebuild the documentation, see Update Documentation.