rocm_jax/docs/jax.rst

jax package
===========

Subpackages
-----------

.. toctree::
    :maxdepth: 1

    jax.numpy
    jax.scipy
    jax.experimental
    jax.lax
    jax.ops
    jax.random
    jax.tree_util

Module contents
---------------

.. automodule:: jax
    :members: jit, disable_jit, grad, value_and_grad, vmap, pmap, jacfwd, jacrev, hessian, jvp, linearize, vjp, make_jaxpr, eval_shape, custom_transforms, defjvp, defjvp_all, defvjp, defvjp_all, custom_gradient, xla_computation
    :undoc-members:
    :show-inheritance:
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00			`jax package`
			`===========`

			`Subpackages`
			`-----------`

			`.. toctree::`
			`:maxdepth: 1`

			`jax.numpy`
			`jax.scipy`
			`jax.experimental`
			`jax.lax`
Add new functions `jax.ops.index_add` and `jax.ops.index_update` for NumPy-style indexed updates. Create a new library `jax.ops` for user-facing ops that don't exist in NumPy or SciPy. Progress on issue #101. Fixes #122. Reenable some disabled TPU indexing tests that now pass. 2019-02-22 07:55:36 -05:00			`jax.ops`
add jax.random entry to jax.rst 2019-02-13 19:31:41 -08:00			`jax.random`
Expose jax.tree_util in the JAX docs. 2019-05-14 21:00:27 -04:00			`jax.tree_util`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
			`Module contents`
			`---------------`

			`.. automodule:: jax`
add xla_computation to jax rst docs 2019-07-05 16:55:11 -07:00			`:members: jit, disable_jit, grad, value_and_grad, vmap, pmap, jacfwd, jacrev, hessian, jvp, linearize, vjp, make_jaxpr, eval_shape, custom_transforms, defjvp, defjvp_all, defvjp, defvjp_all, custom_gradient, xla_computation`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00			`:undoc-members:`
			`:show-inheritance:`