..
|
_static
|
_templates
|
debugging
|
jax-101
|
jep
|
notebooks
|
sphinxext
|
autodidax.ipynb
|
README.md
|
api_compatibility.md
|
async_dispatch.rst
|
autodidax.md
|
autodidax.py
|
changelog.md
|
concurrency.rst
|
conf.py
|
contributing.md
|
deprecation.md
|
developer.md
|
device_memory_profiling.md
|
errors.rst
|
faq.rst
|
glossary.rst
|
gpu_memory_allocation.rst
|
index.rst
|
installation.rst
|
jax.config.rst
|
jax.debug.rst
|
jax.distributed.rst
|
jax.dlpack.rst
|
jax.example_libraries.optimizers.rst
|
jax.example_libraries.rst
|
jax.example_libraries.stax.rst
|
jax.experimental.checkify.rst
|
jax.experimental.global_device_array.rst
|
jax.experimental.host_callback.rst
|
jax.experimental.jet.rst
|
jax.experimental.maps.rst
|
jax.experimental.pjit.rst
|
jax.experimental.rst
|
jax.experimental.sparse.rst
|
jax.flatten_util.rst
|
jax.image.rst
|
jax.lax.rst
|
jax.lib.rst
|
jax.nn.initializers.rst
|
jax.nn.rst
|
jax.numpy.rst
|
jax.ops.rst
|
jax.profiler.rst
|
jax.random.rst
|
jax.rst
|
jax.scipy.rst
|
jax.tree_util.rst
|
jax_internal_api.rst
|
jaxpr.rst
|
multi_process.md
|
profiling.md
|
pytrees.md
|
rank_promotion_warning.rst
|
requirements.txt
|
transfer_guard.rst
|
transformations.md
|
type_promotion.rst
|