File | Mode | Size |
---|---|---|
_static | ||
_templates | ||
notebooks | ||
CHANGELOG.rst | -rw-r--r-- | 17.0 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.5 KB |
developer.rst | -rw-r--r-- | 7.8 KB |
device_memory_profiling.md | -rw-r--r-- | 4.5 KB |
faq.rst | -rw-r--r-- | 12.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.image.rst | -rw-r--r-- | 233 bytes |
jax.lax.rst | -rw-r--r-- | 2.8 KB |
jax.nn.initializers.rst | -rw-r--r-- | 488 bytes |
jax.nn.rst | -rw-r--r-- | 575 bytes |
jax.numpy.rst | -rw-r--r-- | 6.1 KB |
jax.ops.rst | -rw-r--r-- | 1.6 KB |
jax.profiler.rst | -rw-r--r-- | 587 bytes |
jax.random.rst | -rw-r--r-- | 100 bytes |
jax.rst | -rw-r--r-- | 1.8 KB |
jax.scipy.rst | -rw-r--r-- | 2.2 KB |
jax.tree_util.rst | -rw-r--r-- | 110 bytes |
jax_internal_api.rst | -rw-r--r-- | 162 bytes |
jaxpr.rst | -rw-r--r-- | 17.5 KB |
profiling.md | -rw-r--r-- | 10.9 KB |
pytrees.rst | -rw-r--r-- | 8.7 KB |
rank_promotion_warning.rst | -rw-r--r-- | 1.6 KB |
requirements.txt | -rw-r--r-- | 651 bytes |
type_promotion.rst | -rw-r--r-- | 6.9 KB |