Skip to main content

Browse the archive

https://github.com/google/jax
22 May 2024, 15:31:03 UTC
History
Tip revision: 03ed46dcd6dacbaf6d8d3bc2d86a4d15a604be66 authored by Yash Katariya on 29 December 2022, 21:00:06 UTC
Remove `__dict__` from C++ Array. Also disallow setting attributes on Array.
Tip revision: 03ed46d
File Mode Size
_static
_templates
debugging
jax-101
jep
notebooks
sphinxext
Custom_Operation_for_GPUs.md -rw-r--r-- 84.1 KB
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.8 KB
contributing.md -rw-r--r-- 6.3 KB
deprecation.md -rw-r--r-- 554 bytes
developer.md -rw-r--r-- 16.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.5 KB
glossary.rst -rw-r--r-- 4.0 KB
gpu_memory_allocation.rst -rw-r--r-- 2.8 KB
index.rst -rw-r--r-- 2.3 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-- 342 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-- 2.0 KB
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-- 645 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-- 5.1 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.5 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.4 KB
pytrees.md -rw-r--r-- 10.0 KB
rank_promotion_warning.rst -rw-r--r-- 1.6 KB
requirements.txt -rw-r--r-- 420 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

README.md

back to top