Gemfury - jax/experimental · duality-group / jax v0.4.14 - python package

duality-group / jax python

Repository URL to install this package:

Details

jax / jax / experimental
..
array_serialization
compilation_cache
jax2tf
mosaic
sparse
__init__.py
checkify.py
custom_partitioning.py
host_callback.py
jet.py
maps.py
mesh_utils.py
multihost_utils.py
ode.py
pjit.py
profiler.py
rnn.py
serialize_executable.py
shard_map.py
topologies.py
x64_context.py