https://github.com/google/jax
History
Tip revision: fa758bcff564ba7ac99fc13df0185f49a899aafc authored by jax authors on 09 July 2021, 20:06:13 UTC
Merge pull request #7233 from zhangqiaorjc:update_xla_2
Tip revision: fa758bc
File Mode Size
_static
_templates
jax-101
notebooks
sphinxext
README.md -rw-r--r-- 133 bytes
async_dispatch.rst -rw-r--r-- 5.4 KB
autodidax.ipynb -rw-r--r-- 142.2 KB
autodidax.md -rw-r--r-- 101.0 KB
autodidax.py -rw-r--r-- 100.8 KB
changelog.md -rw-r--r-- 33 bytes
concurrency.rst -rw-r--r-- 609 bytes
conf.py -rw-r--r-- 8.6 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-- 10.9 KB
device_memory_profiling.md -rw-r--r-- 4.5 KB
errors.rst -rw-r--r-- 372 bytes
faq.rst -rw-r--r-- 15.2 KB
glossary.rst -rw-r--r-- 3.9 KB
gpu_memory_allocation.rst -rw-r--r-- 2.8 KB
index.rst -rw-r--r-- 1.7 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.pjit.rst -rw-r--r-- 130 bytes
jax.experimental.rst -rw-r--r-- 432 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.4 KB
jax.nn.initializers.rst -rw-r--r-- 488 bytes
jax.nn.rst -rw-r--r-- 589 bytes
jax.numpy.rst -rw-r--r-- 7.3 KB
jax.ops.rst -rw-r--r-- 1.7 KB
jax.profiler.rst -rw-r--r-- 785 bytes
jax.random.rst -rw-r--r-- 819 bytes
jax.rst -rw-r--r-- 2.1 KB
jax.scipy.rst -rw-r--r-- 3.9 KB
jax.tree_util.rst -rw-r--r-- 465 bytes
jax_internal_api.rst -rw-r--r-- 161 bytes
jaxpr.rst -rw-r--r-- 18.6 KB
multi_process.md -rw-r--r-- 6.3 KB
profiling.md -rw-r--r-- 14.8 KB
pytrees.md -rw-r--r-- 8.9 KB
rank_promotion_warning.rst -rw-r--r-- 1.6 KB
requirements.txt -rw-r--r-- 488 bytes
transformations.md -rw-r--r-- 8.5 KB
type_promotion.rst -rw-r--r-- 10.7 KB

README.md

back to top