Revision 6c05add8ae69fb25e09b3bb537ea90a2a3033b95 authored by yashkatariya on 15 November 2022, 20:59:00 UTC, committed by yashkatariya on 15 November 2022, 20:59:00 UTC
1 parent 233bf21
File | Mode | Size |
---|---|---|
_static | ||
_templates | ||
debugging | ||
jax-101 | ||
jep | ||
notebooks | ||
sphinxext | ||
README.md | -rw-r--r-- | 133 bytes |
aot.md | -rw-r--r-- | 9.1 KB |
api_compatibility.md | -rw-r--r-- | 1.8 KB |
async_dispatch.rst | -rw-r--r-- | 5.5 KB |
autodidax.ipynb | -rw-r--r-- | 140.7 KB |
autodidax.md | -rw-r--r-- | 104.0 KB |
autodidax.py | -rw-r--r-- | 103.0 KB |
changelog.md | -rw-r--r-- | 33 bytes |
concurrency.rst | -rw-r--r-- | 609 bytes |
conf.py | -rw-r--r-- | 9.5 KB |
contributing.md | -rw-r--r-- | 6.3 KB |
deprecation.md | -rw-r--r-- | 528 bytes |
developer.md | -rw-r--r-- | 15.2 KB |
device_memory_profiling.md | -rw-r--r-- | 4.5 KB |
errors.rst | -rw-r--r-- | 410 bytes |
faq.rst | -rw-r--r-- | 33.6 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.8 KB |
installation.rst | -rw-r--r-- | 307 bytes |
jax.config.rst | -rw-r--r-- | 388 bytes |
jax.debug.rst | -rw-r--r-- | 624 bytes |
jax.distributed.rst | -rw-r--r-- | 188 bytes |
jax.dlpack.rst | -rw-r--r-- | 170 bytes |
jax.example_libraries.optimizers.rst | -rw-r--r-- | 187 bytes |
jax.example_libraries.rst | -rw-r--r-- | 937 bytes |
jax.example_libraries.stax.rst | -rw-r--r-- | 169 bytes |
jax.experimental.checkify.rst | -rw-r--r-- | 340 bytes |
jax.experimental.global_device_array.rst | -rw-r--r-- | 220 bytes |
jax.experimental.host_callback.rst | -rw-r--r-- | 249 bytes |
jax.experimental.jet.rst | -rw-r--r-- | 126 bytes |
jax.experimental.maps.rst | -rw-r--r-- | 167 bytes |
jax.experimental.pjit.rst | -rw-r--r-- | 130 bytes |
jax.experimental.rst | -rw-r--r-- | 766 bytes |
jax.experimental.sparse.rst | -rw-r--r-- | 464 bytes |
jax.flatten_util.rst | -rw-r--r-- | 218 bytes |
jax.image.rst | -rw-r--r-- | 326 bytes |
jax.lax.rst | -rw-r--r-- | 3.8 KB |
jax.lib.rst | -rw-r--r-- | 658 bytes |
jax.nn.initializers.rst | -rw-r--r-- | 813 bytes |
jax.nn.rst | -rw-r--r-- | 591 bytes |
jax.numpy.rst | -rw-r--r-- | 8.8 KB |
jax.ops.rst | -rw-r--r-- | 484 bytes |
jax.profiler.rst | -rw-r--r-- | 648 bytes |
jax.random.rst | -rw-r--r-- | 879 bytes |
jax.rst | -rw-r--r-- | 1.8 KB |
jax.scipy.rst | -rw-r--r-- | 4.9 KB |
jax.sharding.rst | -rw-r--r-- | 367 bytes |
jax.stages.rst | -rw-r--r-- | 420 bytes |
jax.tree_util.rst | -rw-r--r-- | 465 bytes |
jax_array_migration.md | -rw-r--r-- | 11.2 KB |
jax_internal_api.rst | -rw-r--r-- | 161 bytes |
jaxpr.rst | -rw-r--r-- | 17.3 KB |
multi_process.md | -rw-r--r-- | 8.3 KB |
profiling.md | -rw-r--r-- | 12.5 KB |
pytrees.md | -rw-r--r-- | 10.2 KB |
rank_promotion_warning.rst | -rw-r--r-- | 1.6 KB |
requirements.txt | -rw-r--r-- | 335 bytes |
transfer_guard.rst | -rw-r--r-- | 2.7 KB |
transformations.md | -rw-r--r-- | 8.6 KB |
type_promotion.rst | -rw-r--r-- | 12.7 KB |
![swh spinner](/static/img/swh-spinner.gif)
Computing file changes ...