jax/docs at master · jlebar/jax

Name		Name	Last commit message	Last commit date
parent directory ..
_static		_static
_templates		_templates
README.md		README.md
async_dispatch.rst		async_dispatch.rst
concurrency.rst		concurrency.rst
conf.py		conf.py
gpu_memory_allocation.rst		gpu_memory_allocation.rst
index.rst		index.rst
jax.experimental.optimizers.rst		jax.experimental.optimizers.rst
jax.experimental.rst		jax.experimental.rst
jax.experimental.stax.rst		jax.experimental.stax.rst
jax.lax.rst		jax.lax.rst
jax.numpy.rst		jax.numpy.rst
jax.ops.rst		jax.ops.rst
jax.random.rst		jax.random.rst
jax.rst		jax.rst
jax.scipy.rst		jax.scipy.rst
jax.tree_util.rst		jax.tree_util.rst
modules.rst		modules.rst
profiling.rst		profiling.rst
rank_promotion_warning.rst		rank_promotion_warning.rst
requirements.txt		requirements.txt

README.md

To rebuild the documentation, install the sphinx and sphinx_rtd_theme pip packages and then run:

sphinx-build -M html . build