https://github.com/google/jax
History
Tip revision: 7495f71ea42db7b985427a12a1d366ed3cffda64 authored by jax authors on 03 February 2021, 17:15:38 UTC
Merge pull request #5614 from hawkinsp:jaxlib
Tip revision: 7495f71
File Mode Size
_static
_templates
notebooks
CHANGELOG.rst -rw-r--r-- 26.4 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.6 KB
custom_vjp_update.md -rw-r--r-- 4.9 KB
developer.rst -rw-r--r-- 9.9 KB
device_memory_profiling.md -rw-r--r-- 4.5 KB
faq.rst -rw-r--r-- 13.8 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-- 281 bytes
jax.experimental.loops.rst -rw-r--r-- 136 bytes
jax.experimental.maps.rst -rw-r--r-- 153 bytes
jax.experimental.optimizers.rst -rw-r--r-- 172 bytes
jax.experimental.rst -rw-r--r-- 405 bytes
jax.experimental.stax.rst -rw-r--r-- 154 bytes
jax.image.rst -rw-r--r-- 233 bytes
jax.lax.rst -rw-r--r-- 3.3 KB
jax.nn.initializers.rst -rw-r--r-- 488 bytes
jax.nn.rst -rw-r--r-- 589 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-- 819 bytes
jax.rst -rw-r--r-- 2.0 KB
jax.scipy.rst -rw-r--r-- 3.5 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.4 KB
profiling.md -rw-r--r-- 11.1 KB
pytrees.rst -rw-r--r-- 9.4 KB
rank_promotion_warning.rst -rw-r--r-- 1.6 KB
requirements.txt -rw-r--r-- 651 bytes
type_promotion.rst -rw-r--r-- 10.1 KB

README.md

back to top