|
..
|
|
_static
|
|
_templates
|
|
notebooks
|
|
README.md
|
|
async_dispatch.rst
|
|
concurrency.rst
|
|
conf.py
|
|
developer.rst
|
|
gpu_memory_allocation.rst
|
|
index.rst
|
|
jax.experimental.optimizers.rst
|
|
jax.experimental.rst
|
|
jax.experimental.stax.rst
|
|
jax.experimental.vectorize.rst
|
|
jax.lax.rst
|
|
jax.nn.initializers.rst
|
|
jax.nn.rst
|
|
jax.numpy.rst
|
|
jax.ops.rst
|
|
jax.random.rst
|
|
jax.rst
|
|
jax.scipy.rst
|
|
jax.tree_util.rst
|
|
modules.rst
|
|
profiling.rst
|
|
rank_promotion_warning.rst
|
|
requirements.txt
|