https://github.com/google/jax
History
Tip revision: 6ec90461c68fe8a77cb938ba91b2d0e91725a4b2 authored by jax authors on 05 March 2021, 18:21:11 UTC
Merge pull request #5950 from google:update-pypi
Tip revision: 6ec9046
File Mode Size
_static
_templates
jax-101
notebooks
sphinxext
CHANGELOG.md -rw-r--r-- 25.2 KB
README.md -rw-r--r-- 133 bytes
async_dispatch.rst -rw-r--r-- 5.4 KB
autodidax.ipynb -rw-r--r-- 46.8 KB
autodidax.md -rw-r--r-- 32.3 KB
autodidax.py -rw-r--r-- 32.0 KB
concurrency.rst -rw-r--r-- 700 bytes
conf.py -rw-r--r-- 8.6 KB
custom_vjp_update.md -rw-r--r-- 4.9 KB
developer.md -rw-r--r-- 10.4 KB
device_memory_profiling.md -rw-r--r-- 4.5 KB
errors.rst -rw-r--r-- 328 bytes
faq.rst -rw-r--r-- 15.2 KB
glossary.rst -rw-r--r-- 3.5 KB
gpu_memory_allocation.rst -rw-r--r-- 2.8 KB
index.rst -rw-r--r-- 1.4 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-- 406 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-- 6.2 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.1 KB
jax.scipy.rst -rw-r--r-- 3.8 KB
jax.tree_util.rst -rw-r--r-- 110 bytes
jax_internal_api.rst -rw-r--r-- 161 bytes
jaxpr.rst -rw-r--r-- 18.6 KB
profiling.md -rw-r--r-- 11.1 KB
pytrees.md -rw-r--r-- 8.2 KB
rank_promotion_warning.rst -rw-r--r-- 1.6 KB
requirements.txt -rw-r--r-- 350 bytes
type_promotion.rst -rw-r--r-- 10.1 KB

README.md

back to top