swh:1:snp:71afc25eb6e6e055a37a962e6b91010ec35e397f
History
Tip revision: e27af48510b57e5d9f1ae17f256bd67fa9db7941 authored by jax authors on 06 October 2023, 20:21:49 UTC
Merge pull request #17989 from skye:version
Tip revision: e27af48
File Mode Size
array_serialization
compilation_cache
export
jax2tf
mosaic
pallas
sparse
__init__.py -rw-r--r-- 877 bytes
checkify.py -rw-r--r-- 1.2 KB
custom_partitioning.py -rw-r--r-- 20.9 KB
host_callback.py -rw-r--r-- 77.6 KB
jet.py -rw-r--r-- 26.3 KB
maps.py -rw-r--r-- 964 bytes
mesh_utils.py -rw-r--r-- 15.4 KB
multihost_utils.py -rw-r--r-- 15.7 KB
ode.py -rw-r--r-- 10.4 KB
pjit.py -rw-r--r-- 1.4 KB
profiler.py -rw-r--r-- 1.4 KB
rnn.py -rw-r--r-- 18.9 KB
serialize_executable.py -rw-r--r-- 2.9 KB
shard_map.py -rw-r--r-- 74.5 KB
topologies.py -rw-r--r-- 1.8 KB
x64_context.py -rw-r--r-- 1.7 KB

back to top