| File | Mode | Size |
|---|---|---|
| _static | ||
| _templates | ||
| notebooks | ||
| CHANGELOG.rst | -rw-r--r-- | 12.5 KB |
| README.md | -rw-r--r-- | 133 bytes |
| async_dispatch.rst | -rw-r--r-- | 5.4 KB |
| concurrency.rst | -rw-r--r-- | 700 bytes |
| conf.py | -rw-r--r-- | 9.9 KB |
| developer.rst | -rw-r--r-- | 7.3 KB |
| faq.rst | -rw-r--r-- | 13.2 KB |
| gpu_memory_allocation.rst | -rw-r--r-- | 2.8 KB |
| index.rst | -rw-r--r-- | 1.3 KB |
| jax.dlpack.rst | -rw-r--r-- | 100 bytes |
| jax.experimental.host_callback.rst | -rw-r--r-- | 265 bytes |
| jax.experimental.loops.rst | -rw-r--r-- | 136 bytes |
| jax.experimental.optimizers.rst | -rw-r--r-- | 172 bytes |
| jax.experimental.optix.rst | -rw-r--r-- | 158 bytes |
| jax.experimental.rst | -rw-r--r-- | 263 bytes |
| jax.experimental.stax.rst | -rw-r--r-- | 154 bytes |
| jax.lax.rst | -rw-r--r-- | 2.7 KB |
| jax.nn.initializers.rst | -rw-r--r-- | 488 bytes |
| jax.nn.rst | -rw-r--r-- | 552 bytes |
| jax.numpy.rst | -rw-r--r-- | 4.7 KB |
| jax.ops.rst | -rw-r--r-- | 1.6 KB |
| jax.profiler.rst | -rw-r--r-- | 106 bytes |
| jax.random.rst | -rw-r--r-- | 100 bytes |
| jax.rst | -rw-r--r-- | 1.7 KB |
| jax.scipy.rst | -rw-r--r-- | 2.1 KB |
| jax.tree_util.rst | -rw-r--r-- | 110 bytes |
| jax_internal_api.rst | -rw-r--r-- | 161 bytes |
| jaxpr.rst | -rw-r--r-- | 21.0 KB |
| profiling.rst | -rw-r--r-- | 5.2 KB |
| pytrees.rst | -rw-r--r-- | 8.7 KB |
| rank_promotion_warning.rst | -rw-r--r-- | 1.6 KB |
| requirements.txt | -rw-r--r-- | 246 bytes |
| type_promotion.rst | -rw-r--r-- | 6.9 KB |
