Skip to main content

Browse the archive

https://github.com/google/jax
08 August 2024, 06:23:17 UTC
History
Tip revision: 48ad9a6f3e72f19f5b86f3bb014587e6bc6c0c4c authored by Yash Katariya on 31 May 2023, 23:48:13 UTC
Start jax and jaxlib 0.4.11 release
Tip revision: 48ad9a6
File Mode Size
clusters
debugger
image
internal_test_util
interpreters
lax
lib
nn
numpy
ops
scipy
state
third_party
__init__.py -rw-r--r-- 581 bytes
abstract_arrays.py -rw-r--r-- 3.2 KB
ad_checkpoint.py -rw-r--r-- 33.9 KB
ad_util.py -rw-r--r-- 3.9 KB
api.py -rw-r--r-- 125.0 KB
api_util.py -rw-r--r-- 24.6 KB
array.py -rw-r--r-- 25.9 KB
basearray.py -rw-r--r-- 2.6 KB
basearray.pyi -rw-r--r-- 10.3 KB
callback.py -rw-r--r-- 16.0 KB
checkify.py -rw-r--r-- 50.9 KB
cloud_tpu_init.py -rw-r--r-- 2.3 KB
compilation_cache.py -rw-r--r-- 12.7 KB
compilation_cache_interface.py -rw-r--r-- 831 bytes
config.py -rw-r--r-- 50.3 KB
core.py -rw-r--r-- 110.9 KB
custom_api_util.py -rw-r--r-- 876 bytes
custom_batching.py -rw-r--r-- 10.2 KB
custom_derivatives.py -rw-r--r-- 57.1 KB
custom_transpose.py -rw-r--r-- 8.1 KB
debugging.py -rw-r--r-- 24.9 KB
deprecations.py -rw-r--r-- 1.7 KB
dispatch.py -rw-r--r-- 25.2 KB
distributed.py -rw-r--r-- 7.1 KB
dlpack.py -rw-r--r-- 2.5 KB
dtypes.py -rw-r--r-- 21.6 KB
effects.py -rw-r--r-- 2.3 KB
environment_info.py -rw-r--r-- 1.9 KB
errors.py -rw-r--r-- 20.5 KB
flatten_util.py -rw-r--r-- 3.5 KB
gfile_cache.py -rw-r--r-- 1.9 KB
iree.py -rw-r--r-- 6.0 KB
lax_reference.py -rw-r--r-- 16.9 KB
lazy_loader.py -rw-r--r-- 1.3 KB
linear_util.py -rw-r--r-- 13.4 KB
maps.py -rw-r--r-- 84.3 KB
mesh.py -rw-r--r-- 9.8 KB
monitoring.py -rw-r--r-- 1.9 KB
op_shardings.py -rw-r--r-- 5.0 KB
partition_spec.py -rw-r--r-- 1.6 KB
path.py -rw-r--r-- 1.1 KB
pjit.py -rw-r--r-- 84.1 KB
pretty_printer.py -rw-r--r-- 12.6 KB
prng.py -rw-r--r-- 48.1 KB
profiler.py -rw-r--r-- 13.6 KB
public_test_util.py -rw-r--r-- 9.9 KB
random.py -rw-r--r-- 81.4 KB
sharding.py -rw-r--r-- 4.4 KB
sharding_impls.py -rw-r--r-- 44.0 KB
sharding_specs.py -rw-r--r-- 13.9 KB
source_info_util.py -rw-r--r-- 8.7 KB
stages.py -rw-r--r-- 24.3 KB
test_util.py -rw-r--r-- 40.2 KB
traceback_util.py -rw-r--r-- 7.9 KB
tree_util.py -rw-r--r-- 33.6 KB
typing.py -rw-r--r-- 2.5 KB
util.py -rw-r--r-- 17.4 KB
xla_bridge.py -rw-r--r-- 29.7 KB
back to top