rocm_jax/docs/jax.rst

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

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

.. toctree::
    :maxdepth: 1

    jax.numpy
    jax.scipy
    jax.experimental
    jax.lax
    jax.random

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

.. automodule:: jax
    :members: jit, grad, value_and_grad, vmap, jacfwd, jacrev, hessian, jvp, vjp, make_jaxpr
    :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 jax.random entry to jax.rst 2019-02-13 19:31:41 -08:00			`jax.random`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00
			`Module contents`
			`---------------`

			`.. automodule:: jax`
Add documentation for jax.jvp and jax.vjp. 2019-02-19 22:08:14 -05:00			`:members: jit, grad, value_and_grad, vmap, jacfwd, jacrev, hessian, jvp, vjp, make_jaxpr`
Add Sphinx-generated reference documentation for JAX. 2019-01-15 20:14:19 -05:00			`:undoc-members:`
			`:show-inheritance:`