| File | Mode | Size |
|---|---|---|
| _static | ||
| _templates | ||
| jax-101 | ||
| notebooks | ||
| sphinxext | ||
| README.md | -rw-r--r-- | 133 bytes |
| async_dispatch.rst | -rw-r--r-- | 5.5 KB |
| autodidax.ipynb | -rw-r--r-- | 139.8 KB |
| autodidax.md | -rw-r--r-- | 102.6 KB |
| autodidax.py | -rw-r--r-- | 102.5 KB |
| changelog.md | -rw-r--r-- | 33 bytes |
| concurrency.rst | -rw-r--r-- | 609 bytes |
| conf.py | -rw-r--r-- | 8.9 KB |
| contributing.md | -rw-r--r-- | 6.3 KB |
| custom_vjp_update.md | -rw-r--r-- | 4.9 KB |
| deprecation.md | -rw-r--r-- | 528 bytes |
| developer.md | -rw-r--r-- | 11.0 KB |
| device_memory_profiling.md | -rw-r--r-- | 4.5 KB |
| errors.rst | -rw-r--r-- | 410 bytes |
| faq.rst | -rw-r--r-- | 16.8 KB |
| glossary.rst | -rw-r--r-- | 4.0 KB |
| gpu_memory_allocation.rst | -rw-r--r-- | 2.8 KB |
| index.rst | -rw-r--r-- | 1.6 KB |
| jax.config.rst | -rw-r--r-- | 314 bytes |
| jax.dlpack.rst | -rw-r--r-- | 100 bytes |
| jax.example_libraries.optimizers.rst | -rw-r--r-- | 187 bytes |
| jax.example_libraries.rst | -rw-r--r-- | 199 bytes |
| jax.example_libraries.stax.rst | -rw-r--r-- | 169 bytes |
| jax.experimental.ann.rst | -rw-r--r-- | 166 bytes |
| jax.experimental.host_callback.rst | -rw-r--r-- | 281 bytes |
| jax.experimental.loops.rst | -rw-r--r-- | 136 bytes |
| jax.experimental.maps.rst | -rw-r--r-- | 153 bytes |
| jax.experimental.pjit.rst | -rw-r--r-- | 130 bytes |
| jax.experimental.rst | -rw-r--r-- | 427 bytes |
| jax.experimental.sparse.rst | -rw-r--r-- | 160 bytes |
| jax.flatten_util.rst | -rw-r--r-- | 218 bytes |
| jax.image.rst | -rw-r--r-- | 233 bytes |
| jax.lax.rst | -rw-r--r-- | 3.6 KB |
| jax.lib.rst | -rw-r--r-- | 684 bytes |
| jax.nn.initializers.rst | -rw-r--r-- | 488 bytes |
| jax.nn.rst | -rw-r--r-- | 589 bytes |
| jax.numpy.rst | -rw-r--r-- | 7.4 KB |
| jax.ops.rst | -rw-r--r-- | 2.3 KB |
| jax.profiler.rst | -rw-r--r-- | 785 bytes |
| jax.random.rst | -rw-r--r-- | 819 bytes |
| jax.rst | -rw-r--r-- | 2.3 KB |
| jax.scipy.rst | -rw-r--r-- | 4.0 KB |
| jax.tree_util.rst | -rw-r--r-- | 465 bytes |
| jax_internal_api.rst | -rw-r--r-- | 161 bytes |
| jaxpr.rst | -rw-r--r-- | 17.2 KB |
| multi_process.md | -rw-r--r-- | 6.3 KB |
| profiling.md | -rw-r--r-- | 10.0 KB |
| pytrees.md | -rw-r--r-- | 9.0 KB |
| rank_promotion_warning.rst | -rw-r--r-- | 1.6 KB |
| requirements.txt | -rw-r--r-- | 404 bytes |
| transformations.md | -rw-r--r-- | 8.6 KB |
| type_promotion.rst | -rw-r--r-- | 12.3 KB |
