|
..
|
|
_static
|
|
_templates
|
|
notebooks
|
|
CHANGELOG.rst
|
|
README.md
|
|
async_dispatch.rst
|
|
concurrency.rst
|
|
conf.py
|
|
developer.rst
|
|
device_memory_profiling.md
|
|
faq.rst
|
|
gpu_memory_allocation.rst
|
|
index.rst
|
|
jax.dlpack.rst
|
|
jax.experimental.host_callback.rst
|
|
jax.experimental.loops.rst
|
|
jax.experimental.optimizers.rst
|
|
jax.experimental.optix.rst
|
|
jax.experimental.rst
|
|
jax.experimental.stax.rst
|
|
jax.image.rst
|
|
jax.lax.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
|
|
profiling.md
|
|
pytrees.rst
|
|
rank_promotion_warning.rst
|
|
requirements.txt
|
|
type_promotion.rst
|